python312Packages.ultralytics: 8.3.86 -> 8.3.92 (NixOS#390905)

Author: GaetanLepage

pkgs/development/python-modules/ultralytics/default.nix

@@ -1,25 +1,26 @@
 {
   lib,
+  stdenv,
   buildPythonPackage,
   fetchFromGitHub,
 
   # build-system
   setuptools,
 
   # dependencies
-  scipy,
   matplotlib,
   opencv-python,
+  pandas,
   pillow,
+  psutil,
+  py-cpuinfo,
   pyyaml,
   requests,
+  scipy,
+  seaborn,
   torch,
   torchvision,
   tqdm,
-  psutil,
-  py-cpuinfo,
-  pandas,
-  seaborn,
   ultralytics-thop,
 
   # tests
@@ -30,38 +31,37 @@
 
 buildPythonPackage rec {
   pname = "ultralytics";
-  version = "8.3.86";
+  version = "8.3.92";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "ultralytics";
     repo = "ultralytics";
     tag = "v${version}";
-    hash = "sha256-9z6f/48jQVCR744ojNH+T22+JDg31+WEKWi48k5/GoY=";
+    hash = "sha256-+SwhQVEl7tLrhwuCruVLAtVS3U/RJb0ysIKG3bmuZk4=";
   };
 
-  postPatch = ''
-    substituteInPlace pyproject.toml \
-      --replace-fail "numpy>=1.23.0,<=2.1.1" "numpy"
-  '';
-
   build-system = [ setuptools ];
 
+  pythonRelaxDeps = [
+    "numpy"
+  ];
+
   dependencies = [
-    scipy
     matplotlib
     opencv-python
+    pandas
     pillow
+    psutil
+    py-cpuinfo
     pyyaml
     requests
     scipy
+    scipy
+    seaborn
     torch
     torchvision
     tqdm
-    psutil
-    py-cpuinfo
-    pandas
-    seaborn
     ultralytics-thop
   ];
 
@@ -73,34 +73,50 @@ buildPythonPackage rec {
     onnxruntime
   ];
 
-  # rest of the tests require internet access
-  pytestFlagsArray = [ "tests/test_python.py" ];
-
-  disabledTests = [
-    # also remove the individual tests that require internet
-    "test_model_methods"
-    "test_predict_txt"
-    "test_predict_img"
-    "test_predict_visualize"
-    "test_predict_grey_and_4ch"
-    "test_val"
-    "test_train_scratch"
-    "test_train_pretrained"
-    "test_all_model_yamls"
-    "test_workflow"
-    "test_predict_callback_and_setup"
-    "test_results"
-    "test_labels_and_crops"
-    "test_data_annotator"
-    "test_model_embeddings"
-    "test_yolov10"
-    "test_utils_torchutils"
-    "test_yolo_world"
+  pytestFlagsArray = [
+    # rest of the tests require internet access
+    "tests/test_python.py"
   ];
 
+  disabledTests =
+    [
+      # also remove the individual tests that require internet
+      "test_all_model_yamls"
+      "test_data_annotator"
+      "test_labels_and_crops"
+      "test_model_embeddings"
+      "test_model_methods"
+      "test_predict_callback_and_setup"
+      "test_predict_grey_and_4ch"
+      "test_predict_img"
+      "test_predict_txt"
+      "test_predict_visualize"
+      "test_results"
+      "test_train_pretrained"
+      "test_train_scratch"
+      "test_utils_torchutils"
+      "test_val"
+      "test_workflow"
+      "test_yolo_world"
+      "test_yolov10"
+    ]
+    ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
+      # Fatal Python error: Aborted
+      # onnxruntime/capi/_pybind_state.py", line 32 in <module>
+      "test_utils_benchmarks"
+    ]
+    ++ lib.optionals stdenv.hostPlatform.isDarwin [
+      # Fatal Python error: Aborted
+      # ultralytics/utils/checks.py", line 598 in check_imshow
+      "test_utils_checks"
+
+      # RuntimeError: required keyword attribute 'value' has the wrong type
+      "test_utils_benchmarks"
+    ];
+
   meta = {
     homepage = "https://github.com/ultralytics/ultralytics";
-    changelog = "https://github.com/ultralytics/ultralytics/releases/tag/${src.tag}";
+    changelog = "https://github.com/ultralytics/ultralytics/releases/tag/v${version}";
     description = "Train YOLO models for computer vision tasks";
     mainProgram = "yolo";
     license = lib.licenses.agpl3Only;