Skip to content

Commit ae28aca

Browse files
committed
fix ray test
1 parent 5628a16 commit ae28aca

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

tests/tools/test_process_data.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -88,9 +88,8 @@ class ProcessDataRayTest(DataJuicerTestCaseBase):
8888
def setUp(self):
8989
super().setUp()
9090

91-
self._auto_create_ray_cluster()
92-
93-
self.tmp_dir = tempfile.TemporaryDirectory().name
91+
# self._auto_create_ray_cluster()
92+
self.tmp_dir = f'/workspace/tmp/{self.__class__.__name__}'
9493
if not osp.exists(self.tmp_dir):
9594
os.makedirs(self.tmp_dir)
9695

@@ -129,8 +128,8 @@ def tearDown(self):
129128
import ray
130129
ray.shutdown()
131130

132-
if self.tmp_ray_cluster:
133-
self._close_ray_cluster()
131+
# if self.tmp_ray_cluster:
132+
# self._close_ray_cluster()
134133

135134
def test_ray_image(self):
136135
tmp_yaml_file = osp.join(self.tmp_dir, 'config_0.yaml')

0 commit comments

Comments
 (0)