Change pytest invokation and make ruff happy

NVIDIA · Oct 25, 2024 · 1caf2ad · 1caf2ad
1 parent cba433a
commit 1caf2ad
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -69,8 +69,9 @@ jobs:
           set -eEx
           set -o pipefail
 
-          pytest --dead-fixtures
-          pytest -vv --cov
+          python -m pytest -vv --cov
+
+          python -m pytest --dead-fixtures
 
           # <100% coverage might indicate that some tests are not doing what they should,
           # but realistically, it might too hard to reach, so lower it a bit

diff --git a/tests/test_acceptance.py b/tests/test_acceptance.py
@@ -30,7 +30,7 @@
 from cloudai.schema.test_template.sleep.slurm_command_gen_strategy import SleepSlurmCommandGenStrategy
 from cloudai.schema.test_template.sleep.template import Sleep
 from cloudai.schema.test_template.ucc_test.slurm_command_gen_strategy import UCCTestSlurmCommandGenStrategy
-from cloudai.systems import SlurmSystem, StandaloneSystem
+from cloudai.systems import SlurmSystem
 from cloudai.test_definitions.gpt import GPTCmdArgs, GPTTestDefinition
 from cloudai.test_definitions.grok import GrokCmdArgs, GrokTestDefinition
 from cloudai.test_definitions.nccl import NCCLCmdArgs, NCCLTestDefinition

diff --git a/tests/test_slurm_installer.py b/tests/test_slurm_installer.py
@@ -18,6 +18,7 @@
 from unittest.mock import Mock, patch
 
 import pytest
+
 from cloudai import Installable, InstallStatusResult
 from cloudai.installer.installables import DockerImage, GitRepo, PythonExecutable
 from cloudai.installer.slurm_installer import SlurmInstaller