depot/third_party/nixpkgs/pkgs/development/python-modules/optax/tests.nix

34 lines
514 B
Nix
Raw Normal View History

{ buildPythonPackage
, dm-haiku
, pytest-xdist
, pytestCheckHook
, tensorflow
, tensorflow-datasets
, flax
, optax
}:
buildPythonPackage {
pname = "optax-tests";
inherit (optax) version;
src = optax.testsout;
dontBuild = true;
dontInstall = true;
nativeCheckInputs = [
dm-haiku
pytest-xdist
pytestCheckHook
tensorflow
tensorflow-datasets
flax
];
disabledTestPaths = [
# See https://github.com/deepmind/optax/issues/323
"examples/lookahead_mnist_test.py"
];
}