🐛 fix adapter option

This commit is contained in:
Faylixe
2020-12-08 15:10:32 +01:00
parent 981efbd734
commit 914f4e01ef
3 changed files with 3 additions and 4 deletions

View File

@@ -101,9 +101,8 @@ def test_train():
'-p', 'useless_config.json',
'-d', path
])
raise IOError(f'STDOUT: {result.stdout}')
assert result.exit_code == 0
# assert that model checkpoint was created.
assert os.path.exists(join(path, 'model', 'model.ckpt-10.index'))
assert os.path.exists(join(path, 'model', 'checkpoint'))
assert os.path.exists(join(path, 'model', 'model.ckpt-0.meta'))
assert result.exit_code == 0