refactor

Author: GdoongMathew

src/lightning/pytorch/loggers/utilities.py

@@ -121,14 +121,16 @@ def __init__(self, loggers: Union[Iterable[_T], Mapping[str, _T]] = None):
             super().__init__(loggers.values())
             self._dict = dict(zip(loggers.keys(), range(len(loggers))))
         else:
+            default_dict = {}
+            if isinstance(loggers, _ListMap):
+                default_dict = loggers._dict.copy()
             super().__init__(() if loggers is None else loggers)
-            self._dict: dict = {}
+            self._dict: dict = default_dict
 
     def __eq__(self, other: Any) -> bool:
         list_eq = list.__eq__(self, other)
         if isinstance(other, _ListMap):
-            dict_eq = self._dict == other._dict
-            return list_eq and dict_eq
+            return list_eq and self._dict == other._dict
         return list_eq
 
     def copy(self):

tests/tests_pytorch/loggers/test_utilities.py

@@ -42,13 +42,16 @@ def test_version(tmp_path):
         [1, 2],
         {1, 2},
         range(2),
+        _ListMap({"a": 1, "b": 2}),
     ],
 )
 def test_listmap_init(args):
     """Test initialization with different iterable types."""
     lm = _ListMap(args)
     assert len(lm) == len(args)
     assert isinstance(lm, list)
+    if isinstance(args, _ListMap):
+        assert lm == args
 
 
 def test_listmap_append():