浏览代码

fixed model

Toan Quach 1 年之前
父节点
当前提交
da1664761b

+ 1 - 1
taipy/core/_orchestrator/_orchestrator.py

@@ -70,7 +70,7 @@ class _Orchestrator(_AbstractOrchestrator):
         submission = _SubmissionManagerFactory._build_manager()._create(
             submittable.id,  # type: ignore
             submittable._ID_PREFIX,  # type: ignore
-            submittable.config_id if isinstance(submittable, Scenario) else None,  # type: ignore
+            getattr(submittable, "config_id", None),
         )
 
         jobs = []

+ 2 - 2
taipy/core/submission/_submission_model.py

@@ -50,7 +50,7 @@ class _SubmissionModel(_BaseModel):
             id=data["id"],
             entity_id=data["entity_id"],
             entity_type=data["entity_type"],
-            entity_config_id=_BaseModel._deserialize_attribute(data["entity_config_id"]),
+            entity_config_id=data.get("entity_config_id"),
             job_ids=_BaseModel._deserialize_attribute(data["job_ids"]),
             creation_date=data["creation_date"],
             submission_status=SubmissionStatus._from_repr(data["submission_status"]),
@@ -62,7 +62,7 @@ class _SubmissionModel(_BaseModel):
             self.id,
             self.entity_id,
             self.entity_type,
-            _BaseModel._serialize_attribute(self.entity_config_id),
+            self.entity_config_id,
             _BaseModel._serialize_attribute(self.job_ids),
             self.creation_date,
             repr(self.submission_status),

+ 2 - 2
tests/core/job/test_job_manager.py

@@ -349,8 +349,8 @@ def test_cancel_subsequent_jobs():
     task_3 = Task("task_config_3", {}, print, [dn_4], id="task_3")
 
     # Can't get tasks under 1 scenario due to partial not serializable
-    submission_1 = submission_manager._create("scenario_id", Scenario._ID_PREFIX)
-    submission_2 = submission_manager._create("scenario_id", Scenario._ID_PREFIX)
+    submission_1 = submission_manager._create("scenario_id", Scenario._ID_PREFIX, "scenario_config_id")
+    submission_2 = submission_manager._create("scenario_id", Scenario._ID_PREFIX, "scenario_config_id")
 
     _DataManager._set(dn_1)
     _DataManager._set(dn_2)