From 4e059b3e84b97b0ee2d8f93ceb785a54fa752c07 Mon Sep 17 00:00:00 2001
From: jolaem <jola.mirecka@stfc.ac.uk>
Date: Wed, 5 Jun 2024 13:19:44 +0000
Subject: [PATCH] Fix rescale fail withi floats

Rescale transform first performs numerical conversion, and then drops the floating point to prevent failures when floats sneak in during default processing.
---
 src/caked/dataloader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/caked/dataloader.py b/src/caked/dataloader.py
index a994895..cfccb18 100644
--- a/src/caked/dataloader.py
+++ b/src/caked/dataloader.py
@@ -166,7 +166,7 @@ def process(self, paths: list[str], datatype: str):
         for i in transforms:
             if i.startswith("rescale"):
                 transforms.remove(i)
-                rescale = int(i.split("=")[-1])
+                rescale = int(float(i.split("=")[-1]))
 
         if len(transforms) > 0:
             msg = f"The following transformations are not supported: {transforms}"