Skip to content

Commit 8ce9012

Browse files
chore: ruff
1 parent 0535ac1 commit 8ce9012

File tree

1 file changed

+7
-2
lines changed
  • tests/backend/model_manager/load/model_cache/cached_model

1 file changed

+7
-2
lines changed

tests/backend/model_manager/load/model_cache/cached_model/utils.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,13 +21,18 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
2121
x = x + self.buffer2
2222
return x
2323

24-
is_github_ci = os.getenv('GITHUB_ACTIONS') == 'true'
24+
25+
is_github_ci = os.getenv("GITHUB_ACTIONS") == "true"
2526

2627
parameterize_mps_and_cuda = pytest.mark.parametrize(
2728
("device"),
2829
[
2930
pytest.param(
30-
"mps", marks=pytest.mark.skipif(is_github_ci or not torch.backends.mps.is_available(), reason="MPS is very flaky in CI" if is_github_ci else "MPS is not available.")
31+
"mps",
32+
marks=pytest.mark.skipif(
33+
is_github_ci or not torch.backends.mps.is_available(),
34+
reason="MPS is very flaky in CI" if is_github_ci else "MPS is not available.",
35+
),
3136
),
3237
pytest.param("cuda", marks=pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA is not available.")),
3338
],

0 commit comments

Comments
 (0)