update migrate

Signed-off-by: Kevin Su <pingsutw@apache.org>

Author: pingsutw

flytekit/__init__.py

@@ -99,8 +99,6 @@
 - BlobType
 
 """
-import flytekit.migrate  # isort: skip
-
 import os
 import sys
 from typing import Generator
@@ -163,6 +161,7 @@
     StructuredDatasetTransformerEngine,
     StructuredDatasetType,
 )
+import flytekit.migrate  # isort: skip
 
 
 def current_context() -> ExecutionParameters:

flytekit/core/workflow.py

@@ -1517,7 +1517,7 @@ def wf(name: str = "flyteorg"):
     :param default_options: Default options for the workflow when creating a default launch plan. Currently only
          the labels and annotations are allowed to be set as defaults.
     """
-
+    print("in the workflowwwwww")
     def wrapper(fn: Callable[P, FuncOut]) -> PythonFunctionWorkflow:
         workflow_metadata = WorkflowMetadata(on_failure=failure_policy or WorkflowFailurePolicy.FAIL_IMMEDIATELY)
 

flytekit/migrate/__init__.py

@@ -1,2 +1 @@
 from flytekit.migrate import task, dynamic, workflow  # noqa: F401
-

flytekit/migrate/dynamic.py

@@ -1,9 +1,10 @@
 from typing import Callable, Union
 
+from flyte._task import AsyncFunctionTaskTemplate, P, R
+
 import flytekit
 
 import flytekit.migrate
-from flyte._task import AsyncFunctionTaskTemplate, P, R
 
 
 def dynamic_shim(**kwargs) -> Union[AsyncFunctionTaskTemplate, Callable[P, R]]:

flytekit/migrate/task.py

@@ -75,7 +75,7 @@ def task_shim(
         name="flytekit",
         resources=Resources(cpu=0.8, memory="800Mi"),
         image=image,
-        cache="enabled" if cache else "disable",
+        cache="auto" if cache else "disable",
         plugin_config=plugin_config,
     )
     return env.task(retries=retries, pod_template=pod_template_name or pod_template, docs=docs)

flytekit/migrate/workflow.py

@@ -1,7 +1,7 @@
 import flytekit
-
 from flyte import Image, Resources, TaskEnvironment
 
+
 env = TaskEnvironment(
     name="flytekit",
     resources=Resources(cpu=0.8, memory="800Mi"),

migrate_exmaples/hello.py

@@ -1,7 +1,7 @@
 import logging
-
 from flytekit import ImageSpec, dynamic, task, workflow
 
+
 image = ImageSpec(apt_packages=["vim"], packages=["pandas"])
 
 
@@ -24,6 +24,6 @@ def wf(name: str):
 if __name__ == "__main__":
     import flyte
     flyte.init_from_config(log_level=logging.DEBUG)
-    # run = flyte.with_runcontext(log_level=logging.DEBUG).run(wf, name="flyte")
-    # print(run.name)
-    # print(run.url)
+    run = flyte.with_runcontext(log_level=logging.DEBUG).run(wf, name="flyte")
+    print(run.name)
+    print(run.url)