make lint

Signed-off-by: Yuvraj <[email protected]>
flyteorg · Oct 28, 2021 · 7351644 · 7351644
1 parent e6deaa8
commit 7351644
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git 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
@@ -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",