gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

tests: new checkpoint mxnet test + fix utils (#273)

Browse Source 
* tests: new mxnet test + fix utils

new test added:
- test_restore_checkpoint[tensorflow, mxnet]

fix failed tests in CI
improve utils

* tests: fix comments for mxnet checkpoint test and utils
This commit is contained in:
anabwan
2019-04-07 07:36:44 +03:00
committed by GitHub
parent e1e335a4ef
commit 881f78f45a
3 changed files with 42 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
rl_coach/tests/utils/args_utils.py
View File

@@ -395,9 +395,9 @@ def validate_arg_result(flag, p_valid_params, clres=None, process=None,
results.append(last_step[-1])
time.sleep(1)
assert results[-1] >= Def.Consts.num_hs, \
Def.Consts.ASSERT_MSG.format("bigger than " + Def.Consts.num_hs,
results[-1])
assert int(results[-1]) >= Def.Consts.num_hs, \
Def.Consts.ASSERT_MSG.format("bigger than " +
str(Def.Consts.num_hs), results[-1])
elif flag[0] == "-f" or flag[0] == "--framework":
"""
@@ -445,7 +445,8 @@ def validate_arg_result(flag, p_valid_params, clres=None, process=None,
"""
lst_csv = []
# wait until files created
csv_path = get_csv_path(clres=clres, extra_tries=10)
csv_path = get_csv_path(clres=clres, extra_tries=20,
num_expected_files=int(flag[1]))
assert len(csv_path) > 0, \
Def.Consts.ASSERT_MSG.format("paths are not found", csv_path)
@@ -491,8 +492,8 @@ def validate_arg_result(flag, p_valid_params, clres=None, process=None,
# wait until files created
csv_path = get_csv_path(clres=clres, extra_tries=20)
expected_files = int(flag[1])
assert len(csv_path) >= expected_files, \
Def.Consts.ASSERT_MSG.format(str(expected_files),
num_expected_files = int(flag[1])
assert len(csv_path) >= num_expected_files, \
Def.Consts.ASSERT_MSG.format(str(num_expected_files),
str(len(csv_path)))