From b64a27be6df2e97f0a84acdf12ffa995f0a30e61 Mon Sep 17 00:00:00 2001
From: Riya Singh <riyasingh3909@gmail.com>
Date: Sun, 3 Sep 2023 12:38:07 +0530
Subject: [PATCH] minor fix

---
 deepchem/models/tests/test_mxmnet.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/deepchem/models/tests/test_mxmnet.py b/deepchem/models/tests/test_mxmnet.py
index 0a2d9d9bb4..b805ac270c 100644
--- 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