minor fix

deepchem · Sep 21, 2023 · b64a27b · b64a27b
1 parent 7132029
commit b64a27b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deepchem/models/tests/test_mxmnet.py b/deepchem/models/tests/test_mxmnet.py
@@ -2,6 +2,7 @@
 import torch
 import tempfile
 import numpy as np
+import pytest
 from deepchem.feat.molecule_featurizers import MXMNetFeaturizer
 from deepchem.models.torch_models.mxmnet import MXMNet
 
@@ -11,6 +12,7 @@
 ]
 
 
+@pytest.mark.torch
 def test_mxmnet_regression():
     """
     Test MXMNet class for regression