diff --git a/utils/loggers/wandb/wandb_utils.py b/utils/loggers/wandb/wandb_utils.py
index 66fa8f85ec4e..3f2684a7f3e3 100644
--- a/utils/loggers/wandb/wandb_utils.py
+++ b/utils/loggers/wandb/wandb_utils.py
@@ -157,6 +157,8 @@ def __init__(self, opt, run_id, job_type='Training'):
                         self.data_dict = data_dict
                     else:  # Local .yaml dataset file or .zip file
                         self.data_dict = check_dataset(opt.data)
+                else:
+                    self.data_dict = check_dataset(opt.data)
 
                 self.setup_training(opt)
                 if not self.wandb_artifact_data_dict:
@@ -505,4 +507,4 @@ def all_logging_disabled(highest_level=logging.CRITICAL):
     try:
         yield
     finally:
-        logging.disable(previous_level)
+        logging.disable(previous_level)
\ No newline at end of file