Skip to content

Commit 00e7032

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 998ea3e commit 00e7032

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/tests_pytorch/test_cli.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -564,6 +564,7 @@ def add_arguments_to_parser(self, parser):
564564

565565
def test_adapt_checkpoint_hparams_hook(cleandir):
566566
"""Test that the adapt_checkpoint_hparams hook is called and modifications are applied."""
567+
567568
class AdaptHparamsCLI(LightningCLI):
568569
def add_arguments_to_parser(self, parser):
569570
parser.link_arguments("model.out_dim", "model.hidden_dim", compute_fn=lambda x: x * 2)
@@ -597,6 +598,7 @@ def adapt_checkpoint_hparams(self, subcommand, checkpoint_hparams):
597598

598599
def test_adapt_checkpoint_hparams_hook_empty_dict(cleandir):
599600
"""Test that returning empty dict from adapt_checkpoint_hparams disables checkpoint hyperparameter loading."""
601+
600602
class AdaptHparamsEmptyCLI(LightningCLI):
601603
def add_arguments_to_parser(self, parser):
602604
parser.link_arguments("model.out_dim", "model.hidden_dim", compute_fn=lambda x: x * 2)

0 commit comments

Comments
 (0)