whitelist_for_serdes WorkerStatus enum (#6716)

For https://app.graphite.dev/github/pr/dagster-io/internal/1003
dagster-io · Feb 18, 2022 · 3583858 · 3583858
1 parent 95333ec
commit 3583858
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python_modules/dagster/dagster/core/launcher/base.py b/python_modules/dagster/dagster/core/launcher/base.py
@@ -6,6 +6,7 @@
 from dagster.core.origin import PipelinePythonOrigin
 from dagster.core.storage.pipeline_run import PipelineRun
 from dagster.core.workspace.workspace import IWorkspace
+from dagster.serdes import whitelist_for_serdes
 
 
 class LaunchRunContext(NamedTuple):
@@ -35,6 +36,7 @@ def pipeline_code_origin(self) -> Optional[PipelinePythonOrigin]:
         return self.pipeline_run.pipeline_code_origin
 
 
+@whitelist_for_serdes
 class WorkerStatus(Enum):
     RUNNING = "RUNNING"
     NOT_FOUND = "NOT_FOUND"