tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
python310Packages.treex: Enable more tests
+13
-15
+13
-15
pkgs/development/python-modules/treex/default.nix
+13
-15
pkgs/development/python-modules/treex/default.nix
···
14
14
, rich
15
15
, tensorflow
16
16
, treeo
17
+
, torchmetrics
18
+
, pythonRelaxDepsHook
19
+
, torch
17
20
}:
18
21
19
22
buildPythonPackage rec {
···
34
31
# At the time of writing (2022-03-29), rich is currently at version 11.0.0.
35
32
# The treeo dependency is compatible with a patch, but not marked as such in
36
33
# treex. See https://github.com/cgarciae/treex/issues/68.
37
-
postPatch = ''
38
-
substituteInPlace pyproject.toml \
39
-
--replace 'rich = "^11.2.0"' 'rich = "*"' \
40
-
--replace 'treeo = "^0.0.10"' 'treeo = "*"' \
41
-
--replace 'certifi = "^2021.10.8"' 'certifi = "*"'
42
-
'';
34
+
pythonRelaxDeps = [
35
+
"certifi"
36
+
"flax"
37
+
"rich"
38
+
"treeo"
39
+
];
43
40
44
41
nativeBuildInputs = [
45
42
poetry-core
43
+
pythonRelaxDepsHook
46
44
];
47
45
48
46
buildInputs = [ jaxlib ];
···
54
50
pyyaml
55
51
rich
56
52
treeo
53
+
torch
57
54
];
58
55
59
56
checkInputs = [
···
64
59
keras
65
60
pytestCheckHook
66
61
tensorflow
62
+
torchmetrics
67
63
];
68
64
69
-
pythonImportsCheck = [
70
-
"treex"
71
-
];
72
-
73
-
disabledTestPaths = [
74
-
# Require `torchmetrics` which is not packaged in `nixpkgs`.
75
-
"tests/metrics/test_mean_absolute_error.py"
76
-
"tests/metrics/test_mean_square_error.py"
77
-
];
65
+
pythonImportsCheck = [ "treex" ];
78
66
79
67
meta = with lib; {
80
68
description = "Pytree Module system for Deep Learning in JAX";