Skip to content

Commit

Permalink
add ensure ray stop before starting any tests
Browse files Browse the repository at this point in the history
Signed-off-by: Jack Luar <[email protected]>
  • Loading branch information
luarss committed Oct 31, 2024
1 parent d947e62 commit eb67c27
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 0 deletions.
1 change: 1 addition & 0 deletions tools/AutoTuner/test/ref_file_check.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ def setUp(self):
f" tune --samples 1"
for c in configs
]
subprocess.run(["ray", "stop"], shell=True, check=True)

# Make this a test case
def test_files(self):
Expand Down
1 change: 1 addition & 0 deletions tools/AutoTuner/test/smoke_test_algo_eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@ def setUp(self):
f" --reference {self.reference}"
for a, e in self.matrix
]
subprocess.run(["ray", "stop"], shell=True, check=True)

def make_base(self):
os.chdir(orfs_dir)
Expand Down
1 change: 1 addition & 0 deletions tools/AutoTuner/test/smoke_test_sample_iteration.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ def setUp(self):
f" tune --samples {s} --iterations {i}"
for s, i in self.matrix
]
subprocess.run(["ray", "stop"], shell=True, check=True)


class ASAP7SampleIterationSmokeTest(BaseSampleIterationSmokeTest):
Expand Down
1 change: 1 addition & 0 deletions tools/AutoTuner/test/smoke_test_sweep.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ def setUp(self):
f" --jobs {self.jobs}"
f" sweep"
)
subprocess.run(["ray", "stop"], shell=True, check=True)

def test_sweep(self):
raise NotImplementedError(
Expand Down
1 change: 1 addition & 0 deletions tools/AutoTuner/test/smoke_test_tune.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ def setUp(self):
f" --config {self.config}"
f" tune --samples 5"
)
subprocess.run(["ray", "stop"], shell=True, check=True)

def test_tune(self):
raise NotImplementedError(
Expand Down

0 comments on commit eb67c27

Please sign in to comment.