python3Packages.{rlax,distrax}: fix build (#475277)

2025-12-30 19:51:40 +00:00
parent d0aab5bfa1 958785066e
commit 86a2c2875d
2 changed files with 5 additions and 4 deletions
@@ -36,6 +36,9 @@ buildPythonPackage rec {
    flit-core
  ];

+  pythonRemoveDeps = [
+    "tfp-nightly"
+  ];
  dependencies = [
    absl-py
    chex
@@ -2,10 +2,9 @@
  lib,
  buildPythonPackage,
  fetchFromGitHub,
-  fetchpatch,

  # build-system
-  setuptools,
+  flit-core,

  # dependencies
  absl-py,
@@ -15,7 +14,6 @@
  jax,
  jaxlib,
  numpy,
-  tensorflow-probability,

  # tests
  dm-haiku,
@@ -37,7 +35,7 @@ buildPythonPackage rec {
  };

  build-system = [
-    setuptools
+    flit-core
  ];

  dependencies = [