Skip to content

Commit

Permalink
Fixed lint
Browse files Browse the repository at this point in the history
Signed-off-by: Kevin Su <[email protected]>
  • Loading branch information
pingsutw committed Oct 20, 2021
1 parent a1f6a54 commit dc4b63e
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions tests/flytekit/integration/remote/test_remote.py
Original file line number Diff line number Diff line change
Expand Up @@ -236,10 +236,11 @@ def test_execute_python_workflow_dict_of_string_to_string(flyteclient, flyte_wor

def test_execute_python_workflow_list_of_floats(flyteclient, flyte_workflows_register, flyte_remote_env):
"""Test execution of a @workflow-decorated python function and launchplan that are already registered."""
from mock_flyte_repo.workflows.basic.list_float_wf import my_wf
from mock_flyte_repo.workflows.basic.list_float_wf import concat_list, my_wf

# make sure the task name is the same as the name used during registration
my_wf._name = my_wf.name.replace("mock_flyte_repo.", "")
concat_list._name = concat_list.name.replace("mock_flyte_repo.", "")

remote = FlyteRemote.from_config(PROJECT, "development")
xs: typing.List[float] = [42.24, 999.1, 0.0001]
Expand Down Expand Up @@ -286,11 +287,13 @@ def test_execute_joblib_workflow(flyteclient, flyte_workflows_register, flyte_re
assert output_obj == input_obj


@mock.patch("flytekit.core.node.Node")
def test_execute_with_default_launch_plan(node, flyteclient, flyte_workflows_register):
from mock_flyte_repo.workflows.basic.list_float_wf import my_wf
def test_execute_with_default_launch_plan(flyteclient, flyte_workflows_register):
from mock_flyte_repo.workflows.basic.list_float_wf import concat_list, my_wf

# make sure the task name is the same as the name used during registration
my_wf._name = my_wf.name.replace("mock_flyte_repo.", "")
concat_list._name = concat_list.name.replace("mock_flyte_repo.", "")

node.flyte_entity.name.return_value = node.flyte_entity.name.replace("mock_flyte_repo.", "")
remote = FlyteRemote.from_config(PROJECT, "development")
version = uuid.uuid4().hex[:30] + str(int(time.time()))
xs: typing.List[float] = [42.24, 999.1, 0.0001]
Expand Down

0 comments on commit dc4b63e

Please sign in to comment.