Make mypy happy

lightly-ai · May 22, 2024 · c691301 · c691301
1 parent 3a2b2be
commit c691301
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/utils/test_dist__gather__benchmark_module.py b/tests/utils/test_dist__gather__benchmark_module.py
@@ -37,7 +37,7 @@ def close_torch_distributed() -> Generator[None, None, None]:
 
 
 @pytest.mark.skipif(
-    os.getenv("GITHUB_ACTIONS"),
+    os.getenv("GITHUB_ACTIONS") is not None,
     reason="This test is running in parallel and breaks codecov",
 )
 class TestGatherLayerBenchmarkModule:

diff --git a/tests/utils/test_dist__gather__losses.py b/tests/utils/test_dist__gather__losses.py
@@ -63,7 +63,7 @@ def close_torch_distributed() -> Generator[None, None, None]:
 
 
 @pytest.mark.skipif(
-    os.getenv("GITHUB_ACTIONS"),
+    os.getenv("GITHUB_ACTIONS") is not None,
     reason="This test is running in parallel and breaks codecov",
 )
 class TestGatherLayer_Losses: