From bb0206f35395f92be0c00360a65dfad58e3080d0 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Sat, 26 Oct 2024 13:50:26 -0400 Subject: [PATCH] ci: skip `test_data_equal` on the GPU machine This test crashes on the machine iZ0xih0eykcp6eddga4w5iZ with exit code 1: https://github.com/deepmodeling/deepmd-kit/actions/runs/11533273426/job/32106001782 Signed-off-by: Jinzhe Zeng --- source/tests/consistent/io/test_io.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/source/tests/consistent/io/test_io.py b/source/tests/consistent/io/test_io.py index feafde234d..df81c24ff5 100644 --- a/source/tests/consistent/io/test_io.py +++ b/source/tests/consistent/io/test_io.py @@ -21,6 +21,11 @@ DeepEval, ) +from ...utils import ( + CI, + TEST_DEVICE, +) + infer_path = Path(__file__).parent.parent.parent / "infer" @@ -66,6 +71,7 @@ def tearDown(self): elif Path(ii).is_dir(): shutil.rmtree(ii) + @unittest.skipIf(TEST_DEVICE != "cpu" and CI, "Only test on CPU.") def test_data_equal(self): prefix = "test_consistent_io_" + self.__class__.__name__.lower() for backend_name in ("tensorflow", "pytorch", "dpmodel", "jax"):