Skip to content

Commit cd27aed

Browse files
committed
update migrate folder
Signed-off-by: Kevin Su <pingsutw@apache.org>
1 parent af9247d commit cd27aed

10 files changed

Lines changed: 6 additions & 6 deletions

File tree

flytekit/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@
161161
StructuredDatasetTransformerEngine,
162162
StructuredDatasetType,
163163
)
164-
import flytekit.migrate # isort: skip
164+
import flytekit._migrate # isort: skip
165165

166166

167167
def current_context() -> ExecutionParameters:

flytekit/_migrate/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
from flytekit._migrate import task, dynamic, workflow # noqa: F401
241 Bytes
Binary file not shown.
698 Bytes
Binary file not shown.
4.06 KB
Binary file not shown.
796 Bytes
Binary file not shown.
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,11 @@
44

55
import flytekit
66

7-
import flytekit.migrate
7+
import flytekit._migrate
88

99

1010
def dynamic_shim(**kwargs) -> Union[AsyncFunctionTaskTemplate, Callable[P, R]]:
11-
return flytekit.migrate.task.task_shim(**kwargs)
11+
return flytekit._migrate.task.task_shim(**kwargs)
1212

1313

1414
flytekit.dynamic = dynamic_shim
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def task_shim(
7272
docs = Documentation(description=docs.short_description) if docs else None
7373

7474
env = TaskEnvironment(
75-
name="flytekit",
75+
name="flytekit_task",
7676
resources=Resources(cpu=0.8, memory="800Mi"),
7777
image=image,
7878
cache="auto" if cache else "disable",
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44

55
env = TaskEnvironment(
6-
name="flytekit",
6+
name="flytekit_workflow",
77
resources=Resources(cpu=0.8, memory="800Mi"),
88
image=Image.from_debian_base().with_apt_packages("vim").with_pip_packages("flytekit", "pandas"),
99
)

flytekit/migrate/__init__.py

Lines changed: 0 additions & 1 deletion
This file was deleted.

0 commit comments

Comments
 (0)