Hotfix: Convert "dataType" to a correct enum value

coretex-ai · Jun 14, 2024 · 5eaf27f · 5eaf27f
1 parent 7ff8999
commit 5eaf27f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/coretex/entities/task_run/parameter/base_parameter.py b/coretex/entities/task_run/parameter/base_parameter.py
@@ -29,11 +29,11 @@
 
 class BaseParameter(ABC, Generic[T]):
 
-    def __init__(self, name: str, description: str, value: Optional[T], dataType: ParameterType, required: bool, type: int = 3) -> None:
+    def __init__(self, name: str, description: str, value: Optional[T], dataType: str, required: bool, type: int = 3) -> None:
         self.name = name
         self.description = description
         self.value = value
-        self.dataType = dataType
+        self.dataType = ParameterType(dataType)
         self.required = required
         self.type_ = type
 
@@ -80,7 +80,7 @@ def encode(self) -> Dict[str, Any]:
             "name": self.name,
             "description": self.description,
             "value": self.value,
-            "data_type": self.dataType,
+            "data_type": self.dataType.value,
             "required": self.required
         }