Skip to content

Commit

Permalink
Adjust numpy<2.0.0 compatibility (ultralytics#13906)
Browse files Browse the repository at this point in the history
Signed-off-by: Glenn Jocher <[email protected]>
  • Loading branch information
glenn-jocher authored Jun 23, 2024
1 parent c636fe0 commit 6872028
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 6 deletions.
6 changes: 2 additions & 4 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ classifiers = [

# Required dependencies ------------------------------------------------------------------------------------------------
dependencies = [
"numpy<2.0.0", # temporary patch for compat errors https://github.com/ultralytics/yolov5/actions/runs/9538130424/job/26286956354
"numpy>=1.23.5,<2.0.0", # temporary patch for compat errors https://github.com/ultralytics/yolov5/actions/runs/9538130424/job/26286956354
"matplotlib>=3.3.0",
"opencv-python>=4.6.0",
"pillow>=7.1.2",
Expand All @@ -85,8 +85,6 @@ dependencies = [
[project.optional-dependencies]
dev = [
"ipython",
"check-manifest",
"pre-commit",
"pytest",
"pytest-cov",
"coverage[toml]",
Expand Down Expand Up @@ -119,7 +117,7 @@ logging = [
"dvclive>=2.12.0",
]
extra = [
"hub-sdk>=0.0.5", # Ultralytics HUB
"hub-sdk>=0.0.8", # Ultralytics HUB
"ipython", # interactive notebook
"albumentations>=1.4.6", # training augmentations
"pycocotools>=2.0.7", # COCO mAP
Expand Down
2 changes: 1 addition & 1 deletion ultralytics/engine/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,7 +133,7 @@ def __init__(
# Check if Ultralytics HUB model from https://hub.ultralytics.com
if self.is_hub_model(model):
# Fetch model from HUB
checks.check_requirements("hub-sdk>=0.0.6")
checks.check_requirements("hub-sdk>=0.0.8")
self.session = HUBTrainingSession.create_session(model)
model = self.session.model_file

Expand Down
2 changes: 1 addition & 1 deletion ultralytics/hub/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def login(api_key: str = None, save=True) -> bool:
Returns:
(bool): True if authentication is successful, False otherwise.
"""
checks.check_requirements("hub-sdk>=0.0.6")
checks.check_requirements("hub-sdk>=0.0.8")
from hub_sdk import HUBClient

api_key_url = f"{HUB_WEB_ROOT}/settings?tab=api+keys" # set the redirect URL
Expand Down

0 comments on commit 6872028

Please sign in to comment.