fix formatting issues

Netflix · Oct 4, 2023 · cfcfe02 · cfcfe02
1 parent 206a966
commit cfcfe02
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/metaflow/plugins/argo/argo_workflows.py b/metaflow/plugins/argo/argo_workflows.py
@@ -290,14 +290,13 @@ def _construct_resource_requirements(self, resources):
             "ephemeral-storage": "%sM" % str(resources["disk"]),
         }
         y = {
-            "%s.com/gpu".lower()
-            % resources["gpu_vendor"]: str(resources["gpu"])
+            "%s.com/gpu".lower() % resources["gpu_vendor"]: str(resources["gpu"])
             for _ in [0]
             if resources["gpu"] is not None
         }
         return {
-            'requests': x if KUBERNETES_RESOURCE_REQUESTS else {},
-            'limits': {**x, **y} if KUBERNETES_RESOURCE_REQUESTS else y,
+            "requests": x if KUBERNETES_RESOURCE_REQUESTS else {},
+            "limits": {**x, **y} if KUBERNETES_RESOURCE_REQUESTS else y,
         }
 
 

diff --git a/metaflow/plugins/kubernetes/kubernetes_job.py b/metaflow/plugins/kubernetes/kubernetes_job.py
@@ -317,16 +317,16 @@ def _construct_resource_requirements(self):
             "ephemeral-storage": "%sM" % str(self._kwargs["disk"]),
         }
         y = {
-            "%s.com/gpu".lower()
-            % self._kwargs["gpu_vendor"]: str(self._kwargs["gpu"])
+            "%s.com/gpu".lower() % self._kwargs["gpu_vendor"]: str(self._kwargs["gpu"])
             for _ in [0]
             if self._kwargs["gpu"] is not None
         }
         return {
-            'requests': x if KUBERNETES_RESOURCE_REQUESTS else {},
-            'limits': {**x, **y} if KUBERNETES_RESOURCE_REQUESTS else y,
+            "requests": x if KUBERNETES_RESOURCE_REQUESTS else {},
+            "limits": {**x, **y} if KUBERNETES_RESOURCE_REQUESTS else y,
         }
 
+
 class RunningJob(object):
 
     # State Machine implementation for the lifecycle behavior documented in