fix linter warning

Author: japdubengsub

src/lightning/pytorch/loggers/comet.py

@@ -19,7 +19,8 @@
 import logging
 import os
 from argparse import Namespace
-from typing import TYPE_CHECKING, Any, Dict, Literal, Mapping, Optional, Union
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Any, Literal, Optional, Union
 
 from lightning_utilities.core.imports import RequirementCache
 from torch import Tensor
@@ -261,7 +262,7 @@ def __init__(
         self._project_name: Optional[str] = project
         self._experiment_key: Optional[str] = experiment_key
         self._prefix: Optional[str] = prefix
-        self._kwargs: Dict[str, Any] = kwargs
+        self._kwargs: dict[str, Any] = kwargs
 
         # needs to be set before the first `comet_ml` import
         # because comet_ml imported after another machine learning libraries (Torch)
@@ -318,7 +319,7 @@ def experiment(self) -> comet_experiment:
 
     @override
     @rank_zero_only
-    def log_hyperparams(self, params: Union[Dict[str, Any], Namespace]) -> None:
+    def log_hyperparams(self, params: Union[dict[str, Any], Namespace]) -> None:
         params = _convert_params(params)
         self.experiment.__internal_api__log_parameters__(
             parameters=params,
@@ -394,7 +395,7 @@ def version(self) -> Optional[str]:
         if self._experiment is not None:
             return self._experiment.get_key()
 
-    def __getstate__(self) -> Dict[str, Any]:
+    def __getstate__(self) -> dict[str, Any]:
         state = self.__dict__.copy()
 
         # Save the experiment id in case an experiment object already exists,