fix: hooks of task

Signed-off-by: zjgemi <[email protected]>
deepmodeling · Oct 25, 2024 · 67b5876 · 67b5876
1 parent 7446003
commit 67b5876
Show file tree

Hide file tree

Showing 3 changed files with 457 additions and 3 deletions.
diff --git a/src/dflow/client/__init__.py b/src/dflow/client/__init__.py
@@ -1,4 +1,5 @@
 from .v1alpha1_artifact import V1alpha1Artifact
+from .v1alpha1_dag_task import V1alpha1DAGTask
 from .v1alpha1_lifecycle_hook import V1alpha1LifecycleHook
 from .v1alpha1_parameter import V1alpha1Parameter
 from .v1alpha1_retry_strategy import V1alpha1RetryStrategy
@@ -8,4 +9,4 @@
 
 __all__ = ["V1alpha1Artifact", "V1alpha1LifecycleHook", "V1alpha1Parameter",
            "V1alpha1RetryStrategy", "V1alpha1Sequence", "V1alpha1ValueFrom",
-           "V1alpha1WorkflowStep"]
+           "V1alpha1WorkflowStep", "V1alpha1DAGTask"]