Linting & Pyre changes

pytorch · Mar 28, 2024 · 8e79a51 · 8e79a51
1 parent c6f8516
commit 8e79a51
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 4 deletions.
diff --git a/scripts/kueue_test.py b/scripts/kueue_test.py
diff --git a/torchx/schedulers/__init__.py b/torchx/schedulers/__init__.py
@@ -31,8 +31,7 @@
 
 class SchedulerFactory(Protocol):
     # pyre-fixme: Scheduler opts
-    def __call__(self, session_name: str, **kwargs: object) -> Scheduler:
-        ...
+    def __call__(self, session_name: str, **kwargs: object) -> Scheduler: ...
 
 
 def _defer_load_scheduler(path: str) -> SchedulerFactory:

diff --git a/torchx/schedulers/kueue_scheduler.py b/torchx/schedulers/kueue_scheduler.py
@@ -196,7 +196,7 @@ def app_to_resource(
         "spec": task,
     }
     if annotations is not None:
-        resource["metadata"]["annotations"] = annotations  # pyre-ignore [16]
+        resource["metadata"]["annotations"] = annotations  # pyre-ignore[16]
     return resource
 
 

diff --git a/torchx/schedulers/test/kueue_scheduler_test.py b/torchx/schedulers/test/kueue_scheduler_test.py
@@ -907,7 +907,7 @@ def test_min_replicas(self) -> None:
                 continue
 
             resource["spec"]["minAvailable"] = max(  # pyre-ignore[16]
-                0, resource["spec"]["backoffLimit"] - 1
+                0, resource["spec"]["backoffLimit"] - 1  # pyre-ignore[16]
             )
 
         min_available = resource["spec"]["minAvailable"]