diff --git a/executor/engine/__init__.py b/executor/engine/__init__.py index 5f5468b..2191eea 100644 --- 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 index c056b56..808f402 100644 --- 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' ]