move models file into task make backward compatible

Signed-off-by: Yubo Wang <[email protected]> Signed-off-by: Yubo Wang <[email protected]>
flyteorg · May 13, 2023 · ac8f627 · ac8f627
1 parent a6119a8
commit ac8f627
Show file tree

Hide file tree

Showing 10 changed files with 111 additions and 339 deletions.
diff --git a/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/__init__.py b/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/__init__.py
@@ -10,5 +10,4 @@
    TfJob
 """
 
-from .models import CleanPodPolicy, RestartPolicy
-from .task import PS, Chief, RunPolicy, TfJob, Worker
+from .task import PS, Chief, RunPolicy, TfJob, Worker, RestartPolicy, CleanPodPolicy
diff --git a/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/__init__.py b/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/__init__.py
diff --git a/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/chief.py b/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/chief.py
diff --git a/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/ps.py b/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/ps.py
diff --git a/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/restart_policy.py b/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/restart_policy.py
diff --git a/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/run_policy.py b/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/run_policy.py
diff --git a/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/tensorflow_job.py b/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/tensorflow_job.py
diff --git a/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/worker.py b/plugins/flytekit-kf-tensorflow/flytekitplugins/kftensorflow/models/worker.py