jupyter_scheduler/executors.py

@@ -1,12 +1,12 @@
 import io
 import multiprocessing as mp
 import os
+from pathlib import Path
 import shutil
 import tarfile
 import traceback
 from abc import ABC, abstractmethod
 from functools import lru_cache
-from pathlib import Path
 from typing import Dict, List
 
 import fsspec
@@ -21,11 +21,7 @@
 from jupyter_scheduler.parameterize import add_parameters
 from jupyter_scheduler.scheduler import Scheduler
 from jupyter_scheduler.utils import get_utc_timestamp
-from jupyter_scheduler.workflows import (
-    CreateWorkflow,
-    DescribeWorkflow,
-    DescribeWorkflowDefinition,
-)
+from jupyter_scheduler.workflows import CreateWorkflow, DescribeWorkflow, DescribeWorkflowDefinition
 
 
 class ExecutionManager(ABC):
@@ -247,7 +243,7 @@ def serve_workflow_definition(self):
         ).serve(
             cron=self.model.schedule,
             parameters={
-                "model": create_workflow,
+                "tasks": create_workflow.tasks,
                 "root_dir": self.root_dir,
                 "db_url": self.db_url,
             },