Update .pre-commit-config.yaml (#11009)

* Update .pre-commit-config.yaml

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Update __init__.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update .pre-commit-config.yaml

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Pre-commit updates

* Pre-commit updates

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

utils/loggers/comet/comet_utils.py

@@ -11,28 +11,28 @@ import yaml
 
 logger = logging.getLogger(__name__)
 
-COMET_PREFIX = "comet://"
-COMET_MODEL_NAME = os.getenv("COMET_MODEL_NAME", "yolov5")
-COMET_DEFAULT_CHECKPOINT_FILENAME = os.getenv("COMET_DEFAULT_CHECKPOINT_FILENAME", "last.pt")
+COMET_PREFIX = 'comet://'
+COMET_MODEL_NAME = os.getenv('COMET_MODEL_NAME', 'yolov5')
+COMET_DEFAULT_CHECKPOINT_FILENAME = os.getenv('COMET_DEFAULT_CHECKPOINT_FILENAME', 'last.pt')
 
 
 def download_model_checkpoint(opt, experiment):
-    model_dir = f"{opt.project}/{experiment.name}"
+    model_dir = f'{opt.project}/{experiment.name}'
     os.makedirs(model_dir, exist_ok=True)
 
     model_name = COMET_MODEL_NAME
     model_asset_list = experiment.get_model_asset_list(model_name)
 
     if len(model_asset_list) == 0:
-        logger.error(f"COMET ERROR: No checkpoints found for model name : {model_name}")
+        logger.error(f'COMET ERROR: No checkpoints found for model name : {model_name}')
         return
 
     model_asset_list = sorted(
         model_asset_list,
-        key=lambda x: x["step"],
+        key=lambda x: x['step'],
         reverse=True,
     )
-    logged_checkpoint_map = {asset["fileName"]: asset["assetId"] for asset in model_asset_list}
+    logged_checkpoint_map = {asset['fileName']: asset['assetId'] for asset in model_asset_list}
 
     resource_url = urlparse(opt.weights)
     checkpoint_filename = resource_url.query
@@ -44,22 +44,22 @@ def download_model_checkpoint(opt, experiment):
         checkpoint_filename = COMET_DEFAULT_CHECKPOINT_FILENAME
 
     if asset_id is None:
-        logger.error(f"COMET ERROR: Checkpoint {checkpoint_filename} not found in the given Experiment")
+        logger.error(f'COMET ERROR: Checkpoint {checkpoint_filename} not found in the given Experiment')
         return
 
     try:
-        logger.info(f"COMET INFO: Downloading checkpoint {checkpoint_filename}")
+        logger.info(f'COMET INFO: Downloading checkpoint {checkpoint_filename}')
         asset_filename = checkpoint_filename
 
-        model_binary = experiment.get_asset(asset_id, return_type="binary", stream=False)
-        model_download_path = f"{model_dir}/{asset_filename}"
-        with open(model_download_path, "wb") as f:
+        model_binary = experiment.get_asset(asset_id, return_type='binary', stream=False)
+        model_download_path = f'{model_dir}/{asset_filename}'
+        with open(model_download_path, 'wb') as f:
             f.write(model_binary)
 
         opt.weights = model_download_path
 
     except Exception as e:
-        logger.warning("COMET WARNING: Unable to download checkpoint from Comet")
+        logger.warning('COMET WARNING: Unable to download checkpoint from Comet')
         logger.exception(e)
 
 
@@ -75,9 +75,9 @@ def set_opt_parameters(opt, experiment):
     resume_string = opt.resume
 
     for asset in asset_list:
-        if asset["fileName"] == "opt.yaml":
-            asset_id = asset["assetId"]
-            asset_binary = experiment.get_asset(asset_id, return_type="binary", stream=False)
+        if asset['fileName'] == 'opt.yaml':
+            asset_id = asset['assetId']
+            asset_binary = experiment.get_asset(asset_id, return_type='binary', stream=False)
             opt_dict = yaml.safe_load(asset_binary)
             for key, value in opt_dict.items():
                 setattr(opt, key, value)
@@ -85,11 +85,11 @@ def set_opt_parameters(opt, experiment):
 
     # Save hyperparameters to YAML file
     # Necessary to pass checks in training script
-    save_dir = f"{opt.project}/{experiment.name}"
+    save_dir = f'{opt.project}/{experiment.name}'
     os.makedirs(save_dir, exist_ok=True)
 
-    hyp_yaml_path = f"{save_dir}/hyp.yaml"
-    with open(hyp_yaml_path, "w") as f:
+    hyp_yaml_path = f'{save_dir}/hyp.yaml'
+    with open(hyp_yaml_path, 'w') as f:
         yaml.dump(opt.hyp, f)
     opt.hyp = hyp_yaml_path
 
@@ -113,7 +113,7 @@ def check_comet_weights(opt):
         if opt.weights.startswith(COMET_PREFIX):
             api = comet_ml.API()
             resource = urlparse(opt.weights)
-            experiment_path = f"{resource.netloc}{resource.path}"
+            experiment_path = f'{resource.netloc}{resource.path}'
             experiment = api.get(experiment_path)
             download_model_checkpoint(opt, experiment)
             return True
@@ -140,7 +140,7 @@ def check_comet_resume(opt):
         if opt.resume.startswith(COMET_PREFIX):
             api = comet_ml.API()
             resource = urlparse(opt.resume)
-            experiment_path = f"{resource.netloc}{resource.path}"
+            experiment_path = f'{resource.netloc}{resource.path}'
             experiment = api.get(experiment_path)
             set_opt_parameters(opt, experiment)
             download_model_checkpoint(opt, experiment)