Update distributed.py

interTwin-eu · Jan 9, 2025 · 270474d · 270474d
1 parent 77f8eca
commit 270474d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/itwinai/torch/distributed.py b/src/itwinai/torch/distributed.py
@@ -492,7 +492,7 @@ def local_world_size(self) -> int:
         """
         if torch.cuda.is_available():
             return torch.cuda.device_count()
-        if not "LOCAL_WORLD_SIZE" in os.environ:
+        if "LOCAL_WORLD_SIZE" not in os.environ:
             raise RuntimeError(
             "Could not retrieve local world size as CUDA is unavailable and there is "
             "no 'LOCAL_WORLD_SIZE' environment variable."
@@ -674,7 +674,7 @@ def local_world_size(self) -> int:
         """
         if torch.cuda.is_available():
             return torch.cuda.device_count()
-        if not "LOCAL_WORLD_SIZE" in os.environ:
+        if "LOCAL_WORLD_SIZE" not in os.environ:
             raise RuntimeError(
             "Could not retrieve local world size as CUDA is unavailable and there is "
             "no 'LOCAL_WORLD_SIZE' environment variable."