diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 192a0ff..b8099b9 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -22,7 +22,7 @@ repos:
       - id: detect-private-key
 
   - repo: https://github.com/asottile/pyupgrade
-    rev: v3.8.0
+    rev: v3.10.1
     hooks:
       - id: pyupgrade
         name: Upgrade code
@@ -50,7 +50,7 @@ repos:
         # exclude: "README.md|README.zh-CN.md|CONTRIBUTING.md"
 
   - repo: https://github.com/PyCQA/flake8
-    rev: 6.0.0
+    rev: 6.1.0
     hooks:
       - id: flake8
         name: PEP8
diff --git a/utils/loggers/comet/__init__.py b/utils/loggers/comet/__init__.py
index ad7fa55..c14a5f8 100644
--- a/utils/loggers/comet/__init__.py
+++ b/utils/loggers/comet/__init__.py
@@ -365,15 +365,14 @@ class CometLogger:
         data_dict['path'] = artifact_save_dir
 
         metadata_names = metadata.get('names')
-        if type(metadata_names) == dict:
+        if isinstance(metadata_names, dict):
             data_dict['names'] = {int(k): v for k, v in metadata.get('names').items()}
-        elif type(metadata_names) == list:
+        elif isinstance(metadata_names, list):
             data_dict['names'] = {int(k): v for k, v in zip(range(len(metadata_names)), metadata_names)}
         else:
             raise "Invalid 'names' field in dataset yaml file. Please use a list or dictionary"
 
-        data_dict = self.update_data_paths(data_dict)
-        return data_dict
+        return self.update_data_paths(data_dict)
 
     def update_data_paths(self, data_dict):
         path = data_dict.get('path', '')