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