Commit eea397c5 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent d1c8f9b6
Loading
Loading
Loading
Loading
+3 −9
Original line number Diff line number Diff line
@@ -13,7 +13,6 @@
  numpy,
  optax,
  orbax-checkpoint,
  orbax-export,
  pyyaml,
  rich,
  tensorstore,
@@ -36,14 +35,15 @@

buildPythonPackage (finalAttrs: {
  pname = "flax";
  version = "0.12.6";
  version = "0.12.7";
  pyproject = true;
  __structuredAttrs = true;

  src = fetchFromGitHub {
    owner = "google";
    repo = "flax";
    tag = "v${finalAttrs.version}";
    hash = "sha256-rIDfF9W8cxF0njH4e4uhqURQ0C4N8Boe76u6meMgC34=";
    hash = "sha256-a78KiTsCCARWZvbxz9QKdUKnjkDJGXcPVVJu5rU4m/U=";
  };

  build-system = [
@@ -58,7 +58,6 @@ buildPythonPackage (finalAttrs: {
    numpy
    optax
    orbax-checkpoint
    orbax-export
    pyyaml
    rich
    tensorstore
@@ -78,11 +77,6 @@ buildPythonPackage (finalAttrs: {
    tensorflow
  ];

  pytestFlags = [
    # FutureWarning: In the future `np.object` will be defined as the corresponding NumPy scalar.
    "-Wignore::FutureWarning"
  ];

  disabledTestPaths = [
    # Docs test, needs extra deps + we're not interested in it.
    "docs/_ext/codediff_test.py"