Commit a3743d44 authored by Zejun Lin's avatar Zejun Lin Committed by QuanluZhang
Browse files

fix ci (#433)

* fix ci

* expand time range

* expand time range
parent c550bbce
...@@ -46,7 +46,7 @@ def run(): ...@@ -46,7 +46,7 @@ def run():
nnimanager_log_path = fetch_nni_log_path(EXPERIMENT_URL) nnimanager_log_path = fetch_nni_log_path(EXPERIMENT_URL)
current_trial = 0 current_trial = 0
for _ in range(60): for _ in range(120):
time.sleep(1) time.sleep(1)
tuner_status = read_last_line('naive_test/tuner_result.txt') tuner_status = read_last_line('naive_test/tuner_result.txt')
...@@ -68,7 +68,7 @@ def run(): ...@@ -68,7 +68,7 @@ def run():
current_trial = trial current_trial = trial
print('Trial #%d done' % trial) print('Trial #%d done' % trial)
assert experiment_status, 'Failed to finish in 1 min' assert experiment_status, 'Failed to finish in 2 min'
ss1 = json.load(open('naive_test/search_space.json')) ss1 = json.load(open('naive_test/search_space.json'))
ss2 = json.load(open('naive_test/tuner_search_space.json')) ss2 = json.load(open('naive_test/tuner_search_space.json'))
......
...@@ -61,14 +61,14 @@ def test_builtin_dispatcher(dispatch_type, dispatch_name): ...@@ -61,14 +61,14 @@ def test_builtin_dispatcher(dispatch_type, dispatch_name):
nnimanager_log_path = fetch_nni_log_path(EXPERIMENT_URL) nnimanager_log_path = fetch_nni_log_path(EXPERIMENT_URL)
for _ in range(10): for _ in range(20):
time.sleep(3) time.sleep(3)
# check if experiment is done # check if experiment is done
experiment_status = check_experiment_status(nnimanager_log_path) experiment_status = check_experiment_status(nnimanager_log_path)
if experiment_status: if experiment_status:
break break
assert experiment_status, 'Failed to finish in 30 sec' assert experiment_status, 'Failed to finish in 1 min'
def run(dispatch_type): def run(dispatch_type):
'''test all dispatchers whose type is dispatch_type''' '''test all dispatchers whose type is dispatch_type'''
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment