Reformat using pycharm

utils/loggers/__init__.py

@@ -1,8 +1,7 @@
 # YOLOv5 experiment logging utils
+import torch
 import warnings
 from threading import Thread
-
-import torch
 from torch.utils.tensorboard import SummaryWriter
 
 from utils.general import colorstr, emojis
@@ -14,6 +13,7 @@
 
 try:
     import wandb
+
     assert hasattr(wandb, '__version__')  # verify package import not local dir
 except (ImportError, AssertionError):
     wandb = None

utils/loggers/wandb/log_dataset.py

@@ -1,5 +1,4 @@
 import argparse
-
 import yaml
 
 from wandb_utils import WandbLogger

utils/loggers/wandb/sweep.py

@@ -1,7 +1,6 @@
 import sys
-from pathlib import Path
-
 import wandb
+from pathlib import Path
 
 FILE = Path(__file__).absolute()
 sys.path.append(FILE.parents[2].as_posix())  # add utils/ to path

utils/loggers/wandb/wandb_utils.py

@@ -3,10 +3,9 @@
 import logging
 import os
 import sys
+import yaml
 from contextlib import contextmanager
 from pathlib import Path
-
-import yaml
 from tqdm import tqdm
 
 FILE = Path(__file__).absolute()
@@ -150,21 +149,21 @@ def __init__(self, opt, run_id, job_type='Training'):
                 if not opt.resume:
                     if opt.upload_dataset:
                         self.wandb_artifact_data_dict = self.check_and_upload_dataset(opt)
-                
-                    elif opt.data.endswith('_wandb.yaml'): # When dataset is W&B artifact
+
+                    elif opt.data.endswith('_wandb.yaml'):  # When dataset is W&B artifact
                         with open(opt.data, encoding='ascii', errors='ignore') as f:
                             data_dict = yaml.safe_load(f)
                         self.data_dict = data_dict
-                    else: # Local .yaml dataset file or .zip file
+                    else:  # Local .yaml dataset file or .zip file
                         self.data_dict = check_dataset(opt.data)
-                    
+
                 self.setup_training(opt)
                 # write data_dict to config. useful for resuming from artifacts
                 if not self.wandb_artifact_data_dict:
                     self.wandb_artifact_data_dict = self.data_dict
                 self.wandb_run.config.update({'data_dict': self.wandb_artifact_data_dict},
-                                                 allow_val_change=True)
-                
+                                             allow_val_change=True)
+
             if self.job_type == 'Dataset Creation':
                 self.data_dict = self.check_and_upload_dataset(opt)
 
@@ -179,7 +178,7 @@ def check_and_upload_dataset(self, opt):
         Updated dataset info dictionary where local dataset paths are replaced by WAND_ARFACT_PREFIX links.
         """
         assert wandb, 'Install wandb to upload dataset'
-        config_path = self.log_dataset_artifact(opt.data ,
+        config_path = self.log_dataset_artifact(opt.data,
                                                 opt.single_cls,
                                                 'YOLOv5' if opt.project == 'runs/train' else Path(opt.project).stem)
         print("Created dataset config file ", config_path)
@@ -216,7 +215,7 @@ def setup_training(self, opt):
                                                                                            opt.artifact_alias)
             self.val_artifact_path, self.val_artifact = self.download_dataset_artifact(data_dict.get('val'),
                                                                                        opt.artifact_alias)
-        
+
         if self.train_artifact_path is not None:
             train_path = Path(self.train_artifact_path) / 'data/images/'
             data_dict['train'] = str(train_path)
@@ -326,7 +325,7 @@ def log_dataset_artifact(self, data_file, single_cls, project, overwrite_config=
         if data.get('val'):
             data['val'] = WANDB_ARTIFACT_PREFIX + str(Path(project) / 'val')
         path = Path(data_file).stem
-        path = (path if overwrite_config else path + '_wandb') + '.yaml' # updated data.yaml path
+        path = (path if overwrite_config else path + '_wandb') + '.yaml'  # updated data.yaml path
         data.pop('download', None)
         data.pop('path', None)
         with open(path, 'w') as f: