pyrox.dev
lol
python310Packages.treex: Enable more tests
+13
-15
pkgs/development/python-modules/treex/default.nix
+13
-15
pkgs/development/python-modules/treex/default.nix
···
14
, rich
15
, tensorflow
16
, treeo
17
}:
18
19
buildPythonPackage rec {
···
31
# At the time of writing (2022-03-29), rich is currently at version 11.0.0.
32
# The treeo dependency is compatible with a patch, but not marked as such in
33
# 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
-
'';
40
41
nativeBuildInputs = [
42
poetry-core
43
];
44
45
buildInputs = [ jaxlib ];
···
50
pyyaml
51
rich
52
treeo
53
];
54
55
checkInputs = [
···
59
keras
60
pytestCheckHook
61
tensorflow
62
-
];
63
-
64
-
pythonImportsCheck = [
65
-
"treex"
66
];
67
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
-
];
73
74
meta = with lib; {
75
description = "Pytree Module system for Deep Learning in JAX";
···
14
, rich
15
, tensorflow
16
, treeo
17
+
, torchmetrics
18
+
, pythonRelaxDepsHook
19
+
, torch
20
}:
21
22
buildPythonPackage rec {
···
34
# At the time of writing (2022-03-29), rich is currently at version 11.0.0.
35
# The treeo dependency is compatible with a patch, but not marked as such in
36
# treex. See https://github.com/cgarciae/treex/issues/68.
37
+
pythonRelaxDeps = [
38
+
"certifi"
39
+
"flax"
40
+
"rich"
41
+
"treeo"
42
+
];
43
44
nativeBuildInputs = [
45
poetry-core
46
+
pythonRelaxDepsHook
47
];
48
49
buildInputs = [ jaxlib ];
···
54
pyyaml
55
rich
56
treeo
57
+
torch
58
];
59
60
checkInputs = [
···
64
keras
65
pytestCheckHook
66
tensorflow
67
+
torchmetrics
68
];
69
70
+
pythonImportsCheck = [ "treex" ];
71
72
meta = with lib; {
73
description = "Pytree Module system for Deep Learning in JAX";