From 73516447a4be33642dcc87bb131156b29b5c0c1d Mon Sep 17 00:00:00 2001 From: Yuvraj Date: Thu, 28 Oct 2021 15:54:28 +0530 Subject: [PATCH] make lint Signed-off-by: Yuvraj --- plugins/flytekit-kf-pytorch/tests/test_pytorch_task.py | 7 ++++++- .../flytekit-kf-tensorflow/tests/test_tensorflow_task.py | 4 +--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/plugins/flytekit-kf-pytorch/tests/test_pytorch_task.py b/plugins/flytekit-kf-pytorch/tests/test_pytorch_task.py index 1e796fea45..6d5db78bfd 100644 --- a/plugins/flytekit-kf-pytorch/tests/test_pytorch_task.py +++ b/plugins/flytekit-kf-pytorch/tests/test_pytorch_task.py @@ -5,7 +5,12 @@ def test_pytorch_task(): - @task(task_config=PyTorch(num_workers=10), cache=True, cache_version="1", requests=Resources(cpu="1"),) + @task( + task_config=PyTorch(num_workers=10), + cache=True, + cache_version="1", + requests=Resources(cpu="1"), + ) def my_pytorch_task(x: int, y: str) -> int: return x diff --git a/plugins/flytekit-kf-tensorflow/tests/test_tensorflow_task.py b/plugins/flytekit-kf-tensorflow/tests/test_tensorflow_task.py index 6e3b5474cd..2bdaf747d5 100644 --- a/plugins/flytekit-kf-tensorflow/tests/test_tensorflow_task.py +++ b/plugins/flytekit-kf-tensorflow/tests/test_tensorflow_task.py @@ -6,9 +6,7 @@ def test_tensorflow_task(): @task( - task_config=TfJob( - num_workers=10, num_ps_replicas=1, num_chief_replicas=1 - ), + task_config=TfJob(num_workers=10, num_ps_replicas=1, num_chief_replicas=1), cache=True, requests=Resources(cpu="1"), cache_version="1",