Browse Source

[python] add import SageMaker task (#11121)

JieguangZhou 2 years ago
parent
commit
764a24384f

+ 2 - 0
dolphinscheduler-python/pydolphinscheduler/src/pydolphinscheduler/tasks/__init__.py

@@ -25,6 +25,7 @@ from pydolphinscheduler.tasks.http import Http
 from pydolphinscheduler.tasks.map_reduce import MR
 from pydolphinscheduler.tasks.map_reduce import MR
 from pydolphinscheduler.tasks.procedure import Procedure
 from pydolphinscheduler.tasks.procedure import Procedure
 from pydolphinscheduler.tasks.python import Python
 from pydolphinscheduler.tasks.python import Python
+from pydolphinscheduler.tasks.sagemaker import SageMaker
 from pydolphinscheduler.tasks.shell import Shell
 from pydolphinscheduler.tasks.shell import Shell
 from pydolphinscheduler.tasks.spark import Spark
 from pydolphinscheduler.tasks.spark import Spark
 from pydolphinscheduler.tasks.sql import Sql
 from pydolphinscheduler.tasks.sql import Sql
@@ -45,4 +46,5 @@ __all__ = [
     "Sql",
     "Sql",
     "SubProcess",
     "SubProcess",
     "Switch",
     "Switch",
+    "SageMaker",
 ]
 ]