fix cuda test

deepmodeling · Mar 16, 2024 · cd2be16 · cd2be16
1 parent 0ec6bcd
commit cd2be16
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/tests/pt/model/test_dp_atomic_model.py b/source/tests/pt/model/test_dp_atomic_model.py
@@ -233,4 +233,4 @@ def test_virtual_consistency(self):
                 [1, 1, 0, 1],
             ]
         )
-        np.testing.assert_equal(ret1["mask"], expected_mask)
+        np.testing.assert_equal(to_numpy_array(ret1["mask"]), expected_mask)