From e8a99f43f4ec2c66d701726faa16366d16f0a470 Mon Sep 17 00:00:00 2001
From: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
Date: Mon, 11 Nov 2024 20:10:00 -0500
Subject: [PATCH] also skip lammps dpa-2 tests for CUDA

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
---
 source/lmp/tests/test_lammps_dpa_jax.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/source/lmp/tests/test_lammps_dpa_jax.py b/source/lmp/tests/test_lammps_dpa_jax.py
index 10428b2374..cd295e19f9 100644
--- a/source/lmp/tests/test_lammps_dpa_jax.py
+++ b/source/lmp/tests/test_lammps_dpa_jax.py
@@ -19,6 +19,12 @@
     write_lmp_data,
 )
 
+if os.environ.get("CUDA_VISIBLE_DEVICES", "") != "":
+    pytest.skip(
+        reason="The model is generated with CPU",
+        allow_module_level=True,
+    )
+
 pbtxt_file2 = (
     Path(__file__).parent.parent.parent / "tests" / "infer" / "deeppot-1.pbtxt"
 )