fix: allow None action in ActionWrapper

lavague-ai · Oct 14, 2024 · ddaabc7 · ddaabc7
1 parent 3baf09b
commit ddaabc7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lavague-sdk/lavague/sdk/trajectory/model.py b/lavague-sdk/lavague/sdk/trajectory/model.py
@@ -88,7 +88,11 @@ def from_dict(
     def deserialize_action(cls, values: Dict[str, Any]) -> Dict[str, Any]:
         if "action" in values:
             action_data = values["action"]
-            if not isinstance(action_data, Action) and "action_type" in action_data:
+            if (
+                action_data
+                and not isinstance(action_data, Action)
+                and "action_type" in action_data
+            ):
                 action_class = DEFAULT_PARSER.engine_action_builders.get(
                     action_data["action_type"], Action
                 )