From ae28aca27aac6cf77bb8e1011b02895821295b01 Mon Sep 17 00:00:00 2001 From: "jiangnana.jnn" Date: Tue, 24 Dec 2024 16:29:10 +0800 Subject: [PATCH] fix ray test --- tests/tools/test_process_data.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tests/tools/test_process_data.py b/tests/tools/test_process_data.py index 308bb48f3..1a9ee4d08 100644 --- a/tests/tools/test_process_data.py +++ b/tests/tools/test_process_data.py @@ -88,9 +88,8 @@ class ProcessDataRayTest(DataJuicerTestCaseBase): def setUp(self): super().setUp() - self._auto_create_ray_cluster() - - self.tmp_dir = tempfile.TemporaryDirectory().name + # self._auto_create_ray_cluster() + self.tmp_dir = f'/workspace/tmp/{self.__class__.__name__}' if not osp.exists(self.tmp_dir): os.makedirs(self.tmp_dir) @@ -129,8 +128,8 @@ def tearDown(self): import ray ray.shutdown() - if self.tmp_ray_cluster: - self._close_ray_cluster() + # if self.tmp_ray_cluster: + # self._close_ray_cluster() def test_ray_image(self): tmp_yaml_file = osp.join(self.tmp_dir, 'config_0.yaml')