Skip to content

Commit

Permalink
Merge pull request #30 from alan-turing-institute/29-fix-rescale-floa…
Browse files Browse the repository at this point in the history
…t-fail

Fix rescale fail with floats
  • Loading branch information
mooniean authored Jun 21, 2024
2 parents ea3a275 + 4e059b3 commit 83ff072
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/caked/dataloader.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}"
Expand Down

0 comments on commit 83ff072

Please sign in to comment.