diff --git a/pkgs/development/python-modules/objax/default.nix b/pkgs/development/python-modules/objax/default.nix deleted file mode 100644 index 1909658fe20a..000000000000 --- a/pkgs/development/python-modules/objax/default.nix +++ /dev/null @@ -1,83 +0,0 @@ -{ - lib, - buildPythonPackage, - fetchFromGitHub, - jax, - jaxlib, - keras, - numpy, - parameterized, - pillow, - pytestCheckHook, - pythonOlder, - scipy, - setuptools, - tensorboard, - tensorflow, -}: - -buildPythonPackage rec { - pname = "objax"; - version = "1.8.0"; - pyproject = true; - - disabled = pythonOlder "3.9"; - - src = fetchFromGitHub { - owner = "google"; - repo = "objax"; - tag = "v${version}"; - hash = "sha256-WD+pmR8cEay4iziRXqF3sHUzCMBjmLJ3wZ3iYOD+hzk="; - }; - - patches = [ - # Issue reported upstream: https://github.com/google/objax/issues/270 - ./replace-deprecated-device_buffers.patch - ]; - - build-system = [ setuptools ]; - - # Avoid propagating the dependency on `jaxlib`, see - # https://github.com/NixOS/nixpkgs/issues/156767 - buildInputs = [ jaxlib ]; - - dependencies = [ - jax - numpy - parameterized - pillow - scipy - tensorboard - ]; - - pythonImportsCheck = [ "objax" ]; - - # This is necessary to ignore the presence of two protobufs version (tensorflow is bringing an - # older version). - catchConflicts = false; - - nativeCheckInputs = [ - keras - pytestCheckHook - tensorflow - ]; - - enabledTestPaths = [ "tests/*.py" ]; - - 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 = "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 ]; - # Tests test_syncbn_{0,1,2}d and other tests from tests/parallel.py fail - broken = true; - }; -} diff --git a/pkgs/development/python-modules/objax/replace-deprecated-device_buffers.patch b/pkgs/development/python-modules/objax/replace-deprecated-device_buffers.patch deleted file mode 100644 index fc0fd50a90ce..000000000000 --- a/pkgs/development/python-modules/objax/replace-deprecated-device_buffers.patch +++ /dev/null @@ -1,14 +0,0 @@ -diff --git a/objax/util/util.py b/objax/util/util.py -index c31a356..344cf9a 100644 ---- a/objax/util/util.py -+++ b/objax/util/util.py -@@ -117,7 +117,8 @@ def get_local_devices(): - if _local_devices is None: - x = jn.zeros((jax.local_device_count(), 1), dtype=jn.float32) - sharded_x = map_to_device(x) -- _local_devices = [b.device() for b in sharded_x.device_buffers] -+ device_buffers = [buf.data for buf in sharded_x.addressable_shards] -+ _local_devices = [list(b.devices())[0] for b in device_buffers] - return _local_devices - - diff --git a/pkgs/top-level/python-aliases.nix b/pkgs/top-level/python-aliases.nix index cd0f080d77a3..5e7673c9deae 100644 --- a/pkgs/top-level/python-aliases.nix +++ b/pkgs/top-level/python-aliases.nix @@ -493,6 +493,7 @@ mapAliases ({ ntlm-auth = throw "ntlm-auth has been removed, because it relies on the md4 implementation provided by openssl. Use pyspnego instead."; oauth = throw "oauth has been removed as it is unmaintained"; # added 2025-05-16 oauth2 = throw "oauth2 has been removed as it is unmaintained"; # added 2025-05-16 + objax = throw "objax has been removed because the upstream project was archived."; # Added 2025-10-04 openai-triton = triton; # added 2024-07-18 openai-triton-bin = triton-bin; # added 2024-07-18 openai-triton-cuda = triton-cuda; # added 2024-07-18 diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 6e3aefcf48cf..4dc67b8ebc8e 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -10768,8 +10768,6 @@ self: super: with self; { obfsproxy = callPackage ../development/python-modules/obfsproxy { }; - objax = callPackage ../development/python-modules/objax { }; - objexplore = callPackage ../development/python-modules/objexplore { }; objgraph = callPackage ../development/python-modules/objgraph {