Unverified Commit aeb3f4d4 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

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

parents eebfa206 c24d54a7

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

+57 −41
Original line number Diff line number Diff line 
{

  lib,

  stdenv,

  buildPythonPackage,

  fetchFromGitHub,
@@ -7,19 +8,19 @@ 
  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

  ];



  pytestFlagsArray = [

    # rest of the tests require internet access

  pytestFlagsArray = [ "tests/test_python.py" ];

    "tests/test_python.py"

  ];



  disabledTests = [

  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_txt"

      "test_predict_callback_and_setup"

      "test_predict_grey_and_4ch"

      "test_predict_img"

      "test_predict_txt"

      "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_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;