add dask to launcher job type list

Nanguage · May 10, 2023 · e9cf16d · e9cf16d
1 parent 3109034
commit e9cf16d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/executor/engine/__init__.py b/executor/engine/__init__.py
@@ -1,7 +1,7 @@
 from .core import Engine, EngineSetting
 from .job import LocalJob, ThreadJob, ProcessJob
 
-__version__ = '0.1.7'
+__version__ = '0.1.8'
 
 __all__ = [
     'Engine', 'EngineSetting',

diff --git a/executor/engine/launcher/core.py b/executor/engine/launcher/core.py
@@ -24,8 +24,15 @@
 }
 
 
+try:
+    from ..job.dask import DaskJob
+    job_type_classes['dask'] = DaskJob
+except ImportError:
+    pass
+
+
 JOB_TYPES = T.Literal[
-    'local', 'thread', 'process',
+    'local', 'thread', 'process', 'dask',
     'subprocess', 'webapp'
 ]