Revised relevant code to enable graph input

2026-07-27 23:47:46 +00:00 · 2024-07-24 09:14:20 +00:00
parent 95474d01aa
commit 509b4d7745
4 changed files with 44 additions and 10 deletions
@@ -105,8 +105,9 @@ class ModelExperimentLoaderFromDict(ModelTaskLoader):
                name=model_name,
                description=model_data["description"],
                formulation=model_data["formulation"],
+                architecture=model_data["architecture"],
                variables=model_data["variables"],
-                key=model_name,
+                model_type=model_data["model_type"],
            )
            task_l.append(task)
        return QlibModelExperiment(sub_tasks=task_l)