diff options
author | Fabian Affolter <fabian@affolter-engineering.ch> | 2024-02-20 17:58:36 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-02-20 17:58:36 +0100 |
commit | d44a62d19853330f274f1d52fe6be4b93a36fd4d (patch) | |
tree | a5a593e63ebd5178aa1745b6f8c088ac1dcb7320 /pkgs | |
parent | ff381b50a4e00e67752d8e76dd9fd48615fbe80f (diff) | |
parent | fe85853abeabdd441d4b21c56ab695e4dd312a48 (diff) |
Merge pull request #290167 from fabaff/objax-fix
python311Packages.objax: disable failing tests
Diffstat (limited to 'pkgs')
-rw-r--r-- | pkgs/development/python-modules/objax/default.nix | 21 |
1 files changed, 16 insertions, 5 deletions
diff --git a/pkgs/development/python-modules/objax/default.nix b/pkgs/development/python-modules/objax/default.nix index 24a14b37f98ca..7f2725e9d286b 100644 --- a/pkgs/development/python-modules/objax/default.nix +++ b/pkgs/development/python-modules/objax/default.nix @@ -2,22 +2,26 @@ , buildPythonPackage , fetchFromGitHub , fetchpatch -, jaxlib , jax +, jaxlib +, keras , numpy , parameterized , pillow +, pytestCheckHook +, pythonOlder , scipy +, setuptools , tensorboard -, keras -, pytestCheckHook , tensorflow }: buildPythonPackage rec { pname = "objax"; version = "1.8.0"; - format = "setuptools"; + pyproject = true; + + disabled = pythonOlder "3.9"; src = fetchFromGitHub { owner = "google"; @@ -26,6 +30,10 @@ buildPythonPackage rec { hash = "sha256-WD+pmR8cEay4iziRXqF3sHUzCMBjmLJ3wZ3iYOD+hzk="; }; + nativeBuildInputs = [ + setuptools + ]; + # Avoid propagating the dependency on `jaxlib`, see # https://github.com/NixOS/nixpkgs/issues/156767 buildInputs = [ @@ -62,11 +70,14 @@ buildPythonPackage rec { disabledTests = [ # Test requires internet access for prefetching some weights "test_pretrained_keras_weight_0_ResNet50V2" + # ModuleNotFoundError: No module named 'tree' + "TestResNetV2Pretrained" ]; meta = with lib; { - description = "Objax is a machine learning framework that provides an Object Oriented layer for JAX."; + description = "Machine learning framework that provides an Object Oriented layer for JAX"; homepage = "https://github.com/google/objax"; + changelog = "https://github.com/google/objax/releases/tag/v${version}"; license = licenses.asl20; maintainers = with maintainers; [ ndl ]; }; |