pyrox.dev
lol
python312Packages.ultralytics: 8.3.86 -> 8.3.92 (#390905)
authored by
Gaétan Lepage
and committed by
GitHub
aeb3f4d4
eebfa206
+57
-41
+57
-41
pkgs/development/python-modules/ultralytics/default.nix
+57
-41
pkgs/development/python-modules/ultralytics/default.nix
···
1
1
{
2
2
lib,
3
+
stdenv,
3
4
buildPythonPackage,
4
5
fetchFromGitHub,
5
6
···
7
8
setuptools,
8
9
9
10
# dependencies
10
-
scipy,
11
11
matplotlib,
12
12
opencv-python,
13
+
pandas,
13
14
pillow,
15
+
psutil,
16
+
py-cpuinfo,
14
17
pyyaml,
15
18
requests,
19
+
scipy,
20
+
seaborn,
16
21
torch,
17
22
torchvision,
18
23
tqdm,
19
-
psutil,
20
-
py-cpuinfo,
21
-
pandas,
22
-
seaborn,
23
24
ultralytics-thop,
24
25
25
26
# tests
···
30
31
31
32
buildPythonPackage rec {
32
33
pname = "ultralytics";
33
-
version = "8.3.86";
34
+
version = "8.3.92";
34
35
pyproject = true;
35
36
36
37
src = fetchFromGitHub {
37
38
owner = "ultralytics";
38
39
repo = "ultralytics";
39
40
tag = "v${version}";
40
-
hash = "sha256-9z6f/48jQVCR744ojNH+T22+JDg31+WEKWi48k5/GoY=";
41
+
hash = "sha256-+SwhQVEl7tLrhwuCruVLAtVS3U/RJb0ysIKG3bmuZk4=";
41
42
};
42
43
43
-
postPatch = ''
44
-
substituteInPlace pyproject.toml \
45
-
--replace-fail "numpy>=1.23.0,<=2.1.1" "numpy"
46
-
'';
47
-
48
44
build-system = [ setuptools ];
49
45
46
+
pythonRelaxDeps = [
47
+
"numpy"
48
+
];
49
+
50
50
dependencies = [
51
-
scipy
52
51
matplotlib
53
52
opencv-python
53
+
pandas
54
54
pillow
55
+
psutil
56
+
py-cpuinfo
55
57
pyyaml
56
58
requests
57
59
scipy
60
+
scipy
61
+
seaborn
58
62
torch
59
63
torchvision
60
64
tqdm
61
-
psutil
62
-
py-cpuinfo
63
-
pandas
64
-
seaborn
65
65
ultralytics-thop
66
66
];
67
67
···
73
73
onnxruntime
74
74
];
75
75
76
-
# rest of the tests require internet access
77
-
pytestFlagsArray = [ "tests/test_python.py" ];
78
-
79
-
disabledTests = [
80
-
# also remove the individual tests that require internet
81
-
"test_model_methods"
82
-
"test_predict_txt"
83
-
"test_predict_img"
84
-
"test_predict_visualize"
85
-
"test_predict_grey_and_4ch"
86
-
"test_val"
87
-
"test_train_scratch"
88
-
"test_train_pretrained"
89
-
"test_all_model_yamls"
90
-
"test_workflow"
91
-
"test_predict_callback_and_setup"
92
-
"test_results"
93
-
"test_labels_and_crops"
94
-
"test_data_annotator"
95
-
"test_model_embeddings"
96
-
"test_yolov10"
97
-
"test_utils_torchutils"
98
-
"test_yolo_world"
76
+
pytestFlagsArray = [
77
+
# rest of the tests require internet access
78
+
"tests/test_python.py"
99
79
];
100
80
81
+
disabledTests =
82
+
[
83
+
# also remove the individual tests that require internet
84
+
"test_all_model_yamls"
85
+
"test_data_annotator"
86
+
"test_labels_and_crops"
87
+
"test_model_embeddings"
88
+
"test_model_methods"
89
+
"test_predict_callback_and_setup"
90
+
"test_predict_grey_and_4ch"
91
+
"test_predict_img"
92
+
"test_predict_txt"
93
+
"test_predict_visualize"
94
+
"test_results"
95
+
"test_train_pretrained"
96
+
"test_train_scratch"
97
+
"test_utils_torchutils"
98
+
"test_val"
99
+
"test_workflow"
100
+
"test_yolo_world"
101
+
"test_yolov10"
102
+
]
103
+
++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
104
+
# Fatal Python error: Aborted
105
+
# onnxruntime/capi/_pybind_state.py", line 32 in <module>
106
+
"test_utils_benchmarks"
107
+
]
108
+
++ lib.optionals stdenv.hostPlatform.isDarwin [
109
+
# Fatal Python error: Aborted
110
+
# ultralytics/utils/checks.py", line 598 in check_imshow
111
+
"test_utils_checks"
112
+
113
+
# RuntimeError: required keyword attribute 'value' has the wrong type
114
+
"test_utils_benchmarks"
115
+
];
116
+
101
117
meta = {
102
118
homepage = "https://github.com/ultralytics/ultralytics";
103
-
changelog = "https://github.com/ultralytics/ultralytics/releases/tag/${src.tag}";
119
+
changelog = "https://github.com/ultralytics/ultralytics/releases/tag/v${version}";
104
120
description = "Train YOLO models for computer vision tasks";
105
121
mainProgram = "yolo";
106
122
license = lib.licenses.agpl3Only;