nixpkgs/pkgs/development/python-modules/etils/default.nix

90 lines
1.8 KiB
Nix

{ lib
, buildPythonPackage
, fetchPypi
, pythonOlder
, flit-core
# tests
, chex
, jaxlib
, pytest-subtests
, pytest-xdist
, pytestCheckHook
, yapf
# optional
, jupyter
, mediapy
, numpy
, importlib-resources
, typing-extensions
, zipp
, absl-py
, tqdm
, dm-tree
, jax
, tensorflow
}:
buildPythonPackage rec {
pname = "etils";
version = "1.1.0";
format = "pyproject";
disabled = pythonOlder "3.8";
src = fetchPypi {
inherit pname version;
hash = "sha256-eipJUHeaKB70x+WVriFZkLFcHYxviwonhQCSr1rSxkE=";
};
nativeBuildInputs = [
flit-core
];
passthru.optional-dependencies = rec {
array-types = enp;
eapp = [ absl-py /* FIXME package simple-parsing */ ] ++ epy;
ecolab = [ jupyter numpy mediapy ] ++ enp ++ epy;
edc = epy;
enp = [ numpy ] ++ epy;
epath = [ importlib-resources typing-extensions zipp ] ++ epy;
epy = [ typing-extensions ];
etqdm = [ absl-py tqdm ] ++ epy;
etree = array-types ++ epy ++ enp ++ etqdm;
etree-dm = [ dm-tree ] ++ etree;
etree-jax = [ jax ] ++ etree;
etree-tf = [ tensorflow ] ++ etree;
all = array-types ++ eapp ++ ecolab ++ edc ++ enp ++ epath ++ epy ++ etqdm
++ etree ++ etree-dm ++ etree-jax ++ etree-tf;
};
pythonImportsCheck = [
"etils"
];
nativeCheckInputs = [
chex
jaxlib
pytest-subtests
pytest-xdist
pytestCheckHook
yapf
]
++ passthru.optional-dependencies.all;
disabledTests = [
"test_public_access" # requires network access
];
doCheck = false; # error: infinite recursion encountered
meta = with lib; {
changelog = "https://github.com/google/etils/blob/v${version}/CHANGELOG.md";
description = "Collection of eclectic utils for python";
homepage = "https://github.com/google/etils";
license = licenses.asl20;
maintainers = with maintainers; [ mcwitt ];
};
}