Skip to content

Commit c5e426b

Browse files
committed
update test
1 parent 72c82fb commit c5e426b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tests_pytorch/tuner/test_lr_finder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,6 +410,6 @@ def __init__(self):
410410

411411
model = TestModel()
412412
trainer = Trainer(default_root_dir=tmpdir)
413-
lr_finder = trainer.tuner.lr_find(model=model, num_training=1) # force insufficient data points
413+
lr_finder = trainer.tuner.lr_find(model=model, update_attr=True, num_training=1) # force insufficient data points
414414
assert lr_finder.suggestion() is None
415415
assert model.learning_rate == 0.123 # must remain unchanged because suggestion is not possible

0 commit comments

Comments
 (0)