Skip to content

Commit 805d277

Browse files
committed
add ensure ray stop before starting any tests
Signed-off-by: Jack Luar <[email protected]>
1 parent 84c7602 commit 805d277

File tree

5 files changed

+5
-0
lines changed

5 files changed

+5
-0
lines changed

tools/AutoTuner/test/ref_file_check.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ def setUp(self):
2626
f" tune --samples 1"
2727
for c in configs
2828
]
29+
subprocess.run(["ray", "stop"], shell=True, check=True)
2930

3031
# Make this a test case
3132
def test_files(self):

tools/AutoTuner/test/smoke_test_algo_eval.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@ def setUp(self):
3232
f" --reference {self.reference}"
3333
for a, e in self.matrix
3434
]
35+
subprocess.run(["ray", "stop"], shell=True, check=True)
3536

3637
def make_base(self):
3738
os.chdir(orfs_dir)

tools/AutoTuner/test/smoke_test_sample_iteration.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ def setUp(self):
2727
f" tune --samples {s} --iterations {i}"
2828
for s, i in self.matrix
2929
]
30+
subprocess.run(["ray", "stop"], shell=True, check=True)
3031

3132

3233
class ASAP7SampleIterationSmokeTest(BaseSampleIterationSmokeTest):

tools/AutoTuner/test/smoke_test_sweep.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ def setUp(self):
3939
f" --jobs {self.jobs}"
4040
f" sweep"
4141
)
42+
subprocess.run(["ray", "stop"], shell=True, check=True)
4243

4344
def test_sweep(self):
4445
raise NotImplementedError(

tools/AutoTuner/test/smoke_test_tune.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ def setUp(self):
2525
f" --config {self.config}"
2626
f" tune --samples 5"
2727
)
28+
subprocess.run(["ray", "stop"], shell=True, check=True)
2829

2930
def test_tune(self):
3031
raise NotImplementedError(

0 commit comments

Comments
 (0)