33afbf39f6
checkInputs used to be added to nativeBuildInputs. Now we have nativeCheckInputs to do that instead. Doing this treewide change allows to keep hashes identical to before the introduction of nativeCheckInputs.
79 lines
1.3 KiB
Nix
79 lines
1.3 KiB
Nix
{ buildPythonPackage
|
|
, cloudpickle
|
|
, dm-haiku
|
|
, einops
|
|
, fetchFromGitHub
|
|
, flax
|
|
, hypothesis
|
|
, jaxlib
|
|
, keras
|
|
, lib
|
|
, poetry-core
|
|
, pytestCheckHook
|
|
, pyyaml
|
|
, rich
|
|
, tensorflow
|
|
, treeo
|
|
, torchmetrics
|
|
, pythonRelaxDepsHook
|
|
, torch
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "treex";
|
|
version = "0.6.11";
|
|
format = "pyproject";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "cgarciae";
|
|
repo = pname;
|
|
rev = "refs/tags/${version}";
|
|
hash = "sha256-ObOnbtAT4SlrwOms1jtn7/XKZorGISGY6VuhQlC3DaQ=";
|
|
};
|
|
|
|
# At the time of writing (2022-03-29), rich is currently at version 11.0.0.
|
|
# The treeo dependency is compatible with a patch, but not marked as such in
|
|
# treex. See https://github.com/cgarciae/treex/issues/68.
|
|
pythonRelaxDeps = [
|
|
"certifi"
|
|
"flax"
|
|
"rich"
|
|
"treeo"
|
|
];
|
|
|
|
nativeBuildInputs = [
|
|
poetry-core
|
|
pythonRelaxDepsHook
|
|
];
|
|
|
|
buildInputs = [ jaxlib ];
|
|
|
|
propagatedBuildInputs = [
|
|
einops
|
|
flax
|
|
pyyaml
|
|
rich
|
|
treeo
|
|
torch
|
|
];
|
|
|
|
nativeCheckInputs = [
|
|
cloudpickle
|
|
dm-haiku
|
|
hypothesis
|
|
keras
|
|
pytestCheckHook
|
|
tensorflow
|
|
torchmetrics
|
|
];
|
|
|
|
pythonImportsCheck = [ "treex" ];
|
|
|
|
meta = with lib; {
|
|
description = "Pytree Module system for Deep Learning in JAX";
|
|
homepage = "https://github.com/cgarciae/treex";
|
|
license = licenses.mit;
|
|
maintainers = with maintainers; [ ndl ];
|
|
};
|
|
}
|