From a785341db084211c6389ff836bbcb011c3fa1b9b Mon Sep 17 00:00:00 2001
From: Alexander Barannikov <japdubengsub@gmail.com>
Date: Tue, 8 Oct 2024 22:03:58 +0100
Subject: [PATCH] fix cli arguments unit test

---
 tests/tests_pytorch/test_cli.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/tests/tests_pytorch/test_cli.py b/tests/tests_pytorch/test_cli.py
index 56b58d4d157a1..d28e23d5bf87b 100644
--- 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
+        },
     )