fix: skip GPU-dependent tests in CI and fix conftest loading

Author: OuyangWenyu

.github/workflows/build.yml

@@ -91,4 +91,4 @@ jobs:
 #
             - name: PKG-TEST
               run: |
-                  uv run pytest --noconftest -m "not requires_data" tests
+                  uv run pytest -m "not requires_data and not requires_gpu" tests

.github/workflows/docs.yml

@@ -31,7 +31,7 @@ jobs:
             
             - name: Run tests
               run: |
-                  uv run pytest tests
+                  uv run pytest -m "not requires_data and not requires_gpu" tests
             
             - name: Build and deploy docs
               run: |

pyproject.toml

@@ -157,6 +157,7 @@ collect_ignore = []  # 之前setup.py已删除，保留配置项以备将来使
 markers = [
     "slow: marks tests as slow (deselect with '-m \"not slow\"')",
     "requires_data: marks tests that require downloading external data",
+    "requires_gpu: marks tests that require GPU/CUDA (deselect with '-m \"not requires_gpu\"')",
 ]
 
 # black 代码格式化配置

tests/test_cudnnlstm.py

@@ -14,6 +14,7 @@
 from torchhydro.models.cudnnlstm import CudnnLstmModel, CudnnLstm
 
 
+@pytest.mark.requires_gpu
 def test_mc_dropout_eval():
     # Monte Carlo Dropout sampling during evaluation
     model = CudnnLstmModel(
@@ -128,6 +129,7 @@ def test_reset_mask_with_zero_dropout():
     ), "mask_w_hh should have the same shape as w_hh"
 
 
+@pytest.mark.requires_gpu
 def test_forward_with_mc_dropout():
     model = CudnnLstm(input_size=10, hidden_size=20, dr=0.5)
     model = model.to("cuda:0")
@@ -141,6 +143,7 @@ def test_forward_with_mc_dropout():
     assert cy.shape == (1, 5, 20), "Cell state shape mismatch"
 
 
+@pytest.mark.requires_gpu
 def test_forward_with_dropout():
     model = CudnnLstm(input_size=10, hidden_size=20, dr=0.5)
     input_data = torch.randn(20, 5, 10)  # [seq_len, batch_size, input_size]
@@ -154,6 +157,7 @@ def test_forward_with_dropout():
     assert cy.shape == (1, 5, 20), "Cell state shape mismatch"
 
 
+@pytest.mark.requires_gpu
 def test_forward_with_zero_dropout():
     model = CudnnLstm(input_size=10, hidden_size=20, dr=0.0)
     input_data = torch.randn(20, 5, 10)  # [seq_len, batch_size, input_size]
@@ -167,6 +171,7 @@ def test_forward_with_zero_dropout():
     assert cy.shape == (1, 5, 20), "Cell state shape mismatch"
 
 
+@pytest.mark.requires_gpu
 def test_forward_with_dropout_false():
     model = CudnnLstm(input_size=10, hidden_size=20, dr=0.5)
     input_data = torch.randn(20, 5, 10)  # [seq_len, batch_size, input_size]
@@ -180,6 +185,7 @@ def test_forward_with_dropout_false():
     assert cy.shape == (1, 5, 20), "Cell state shape mismatch"
 
 
+@pytest.mark.requires_gpu
 def test_forward_with_initial_states():
     model = CudnnLstm(input_size=10, hidden_size=20, dr=0.5)
     input_data = torch.randn(20, 5, 10)  # [seq_len, batch_size, input_size]

tests/test_train_camelspro_mtl.py

@@ -16,6 +16,7 @@
 from torchhydro.trainers.trainer import train_and_evaluate
 
 
+@pytest.mark.requires_gpu
 def test_cuda_available():
     assert torch.cuda.is_available()