Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix hparams issue

This commit is contained in:
William Falcon 2020-04-26 17:34:33 -04:00
parent d40b4c8f80
commit e7a0d98277
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/trainer/test_trainer.py
View File

@ -27,7 +27,7 @@ from tests.base import (
def test_hparams_save_load(tmpdir):
model = DictHparamsModel({'in_features': 28 * 28, 'out_features': 10})
model = DictHparamsModel({'in_features': 28 * 28, 'out_features': 10, 'failed_key': lambda x: x})
# logger file to get meta
trainer_options = dict(