[BUG] Fix command numeric conversion

compneurobilbao · Sep 20, 2024 · f7d3dde · f7d3dde
1 parent 6d51798
commit f7d3dde
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/ageml/utils.py b/src/ageml/utils.py
@@ -28,7 +28,11 @@ def convert(value):
         converted_value = False
     else:
         try:
-            converted_value = float(value)
+            # If there is . treat as float if not as int
+            if "." in value:
+                converted_value = float(value)
+            else:
+                converted_value = int(value)
         except ValueError:
             # If the value cannot be converted to a float, keep it as a string
             converted_value = value