optimize imports

Signed-off-by: Yee Hing Tong <[email protected]>

flytekit/common/tasks/sagemaker/hpo_job_task.py

@@ -12,7 +12,6 @@
 from flytekit.common.tasks.sagemaker.built_in_training_job_task import SdkBuiltinAlgorithmTrainingJobTask
 from flytekit.common.tasks.sagemaker.custom_training_job_task import CustomTrainingJobTask
 from flytekit.common.tasks.sagemaker.types import HyperparameterTuningJobConfig, ParameterRange
-from flytekit.models.admin import task as _task_models
 from flytekit.models.core import interface as _interface_model
 from flytekit.models.core import literals as _literal_models
 from flytekit.models.core import types as _core_types

flytekit/core/shim_task.py

@@ -6,7 +6,6 @@
 from flytekit import ExecutionParameters, FlyteContext, FlyteContextManager, logger
 from flytekit.core.tracker import TrackedInstance
 from flytekit.core.type_engine import TypeEngine
-from flytekit.models.admin import task as _task_model
 from flytekit.models.core import dynamic_job as _dynamic_job
 from flytekit.models.core import literals as _literal_models
 

flytekit/extras/sqlite3/task.py

@@ -14,7 +14,6 @@
 from flytekit.core.context_manager import SerializationSettings
 from flytekit.core.python_customized_container_task import PythonCustomizedContainerTask
 from flytekit.core.shim_task import ShimTaskExecutor
-from flytekit.models.admin import task as task_models
 from flytekit.types.schema import FlyteSchema
 
 

flytekit/models/core/dynamic_job.py

@@ -2,7 +2,6 @@
 
 import flytekit.models.core.task
 from flytekit.models import common as _common
-from flytekit.models.admin import task as _task
 from flytekit.models.core import literals as _literals
 from flytekit.models.core import workflow as _workflow
 

flytekit/models/core/workflow_closure.py

@@ -2,7 +2,6 @@
 
 import flytekit.models.core.task
 from flytekit.models import common as _common
-from flytekit.models.admin import task as _task_models
 from flytekit.models.core import workflow as _core_workflow_models
 
 

flytekit/remote/component_nodes.py

@@ -1,11 +1,9 @@
 import logging as _logging
 from typing import Dict
 
-import flytekit
 import flytekit.models.core.task
 from flytekit.common.exceptions import system as _system_exceptions
 from flytekit.models.admin import launch_plan as _launch_plan_model
-from flytekit.models.admin import task as _task_model
 from flytekit.models.core import workflow as _workflow_model
 from flytekit.remote import identifier as _identifier
 

flytekit/remote/tasks/task.py

@@ -5,7 +5,6 @@
 from flytekit.core.interface import Interface
 from flytekit.core.type_engine import TypeEngine
 from flytekit.loggers import logger
-from flytekit.models.admin import task as _task_model
 from flytekit.models.core import identifier as _identifier_model
 from flytekit.remote import identifier as _identifier
 from flytekit.remote import interface as _interfaces

flytekit/remote/workflow.py

@@ -8,7 +8,6 @@
 from flytekit.core.interface import Interface
 from flytekit.core.type_engine import TypeEngine
 from flytekit.models.admin import launch_plan as _launch_plan_models
-from flytekit.models.admin import task as _task_models
 from flytekit.models.core import identifier as _identifier_model
 from flytekit.models.core import workflow as _workflow_models
 from flytekit.remote import identifier as _identifier

tests/flytekit/unit/models/test_common.py

@@ -1,6 +1,5 @@
 import flytekit.models.admin.common
 import flytekit.models.admin.launch_plan
-from flytekit.models import common as _common
 from flytekit.models.admin import common as _admin_common
 from flytekit.models.core import execution as _execution