HUB CI fix (#1776)

2 years ago · 4198570a4b
parent 84948651cd
commit 4198570a4b
2 changed files with 2 additions and 2 deletions
--- a/ultralytics/init.py
+++ b/ultralytics/init.py
@ -1,6 +1,6 @@
 # Ultralytics YOLO 🚀, GPL-3.0 license

-__version__ = '8.0.60'
+__version__ = '8.0.61'

 from ultralytics.hub import start
 from ultralytics.yolo.engine.model import YOLO
--- 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