SIENTIAPDE-994

Refactor activity preparation methods and update Docker configuration for simulator
2025-05-13 10:17:39 -03:00
parent d22545ba88
commit 1156037e08
10 changed files with 290 additions and 11 deletions
--- a/laborious/activities/activities.py
+++ b/laborious/activities/activities.py
@@ -50,5 +50,5 @@ class Activities(Postgres, MLFlow, Gates, OPC):
                     notification_handler=notification_handler)

    @activity.defn(name="prepare_activity")
-    async def prepare_activity(self, schedule_name: str, model_name: str, model_id: str):
-        await super().prepare_activity(schedule_name, model_name, model_id)
+    def prepare_activity(self, input_data: dict[str, Any]):
+        super().prepare_activity(input_data)