fix(steps): better int parsing...

needs to be able to parse a string to a float and then an int
pollination · Dec 19, 2020 · 23354f8 · 23354f8
1 parent a017cdb
commit 23354f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/queenbee/io/inputs/step.py b/queenbee/io/inputs/step.py
@@ -158,7 +158,7 @@ def from_template(template: Union[DAGInputs, FunctionInputs], value: Any) -> Ste
         return StepStringInput.parse_obj(template_dict)
 
     if template.__class__ in [DAGIntegerInput, FunctionIntegerInput]:
-        template_dict['value'] = int(template_dict['value'])
+        template_dict['value'] = int(float(template_dict['value']))
         return StepIntegerInput.parse_obj(template_dict)
 
     if template.__class__ in [DAGNumberInput, FunctionNumberInput]:

diff --git a/queenbee/io/outputs/step.py b/queenbee/io/outputs/step.py
@@ -131,7 +131,7 @@ def from_template(template: Union[DAGOutputs, FunctionOutputs], value: Any) -> S
         return StepStringOutput.parse_obj(template_dict)
 
     if template.__class__ in [DAGIntegerOutput, FunctionIntegerOutput]:
-        template_dict['value'] = int(template_dict['value'])
+        template_dict['value'] = int(float(template_dict['value']))
         return StepIntegerOutput.parse_obj(template_dict)
 
     if template.__class__ in [DAGNumberOutput, FunctionNumberOutput]: