add algorithm details

causy-dev · Jul 6, 2024 · 0ec9210 · 0ec9210
1 parent 827e322
commit 0ec9210
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/causy/serialization.py b/causy/serialization.py
@@ -31,9 +31,12 @@ def load_algorithm_from_specification(algorithm_dict: Dict[str, Any]):
     algorithm_dict["pipeline_steps"] = load_pipeline_steps_by_definition(
         algorithm_dict["pipeline_steps"]
     )
-    algorithm_dict["extensions"] = load_pipeline_steps_by_definition(
-        algorithm_dict["extensions"]
-    )
+    if "extensions" in algorithm_dict and algorithm_dict["extensions"] is not None:
+        algorithm_dict["extensions"] = load_pipeline_steps_by_definition(
+            algorithm_dict["extensions"]
+        )
+    else:
+        algorithm_dict["extensions"] = []
     if "variables" not in algorithm_dict or algorithm_dict["variables"] is None:
         algorithm_dict["variables"] = []