fix cli arguments unit test

Lightning-AI · Oct 8, 2024 · a785341 · a785341
1 parent 512d763
commit a785341
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/tests_pytorch/test_cli.py b/tests/tests_pytorch/test_cli.py
@@ -1580,8 +1580,13 @@ def _test_logger_init_args(logger_name, init, unresolved=None):
 def test_comet_logger_init_args():
  _test_logger_init_args(
  "CometLogger",
- init={"save_dir": "comet"}, # Resolve from CometLogger.__init__
- unresolved={"workspace": "comet"}, # Resolve from Comet{,Existing,Offline}Experiment.__init__
+ init={
+ "experiment_key": "some_key", # Resolve from CometLogger.__init__
+ "workspace": "comet",
+ },
+ unresolved={
+ "save_dir": "comet", # Resolve from CometLogger.__init__ as kwarg
+ },
  )