HUB CI fix (#1776)

2023-04-03 03:01:16 +02:00
parent 84948651cd
commit 4198570a4b
2 changed files with 2 additions and 2 deletions
--- a/ultralytics/yolo/engine/model.py
+++ b/ultralytics/yolo/engine/model.py
@ -92,7 +92,7 @@ class YOLO:

        # Check if Ultralytics HUB model from https://hub.ultralytics.com
        if model.startswith('https://hub.ultralytics.com/models/'):
-            from ultralytics.hub import HUBTrainingSession
+            from ultralytics.hub.session import HUBTrainingSession
            self.session = HUBTrainingSession(model)
            model = self.session.model_file