Unverified Commit 52a81336 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.jax: 0.8.2 -> 0.9.2 (#482118)

parents 67a9f0ae 38cc1f18

pkgs/development/python-modules/bambi/default.nix

+8 −0
Original line number Diff line number Diff line
@@ -36,6 +36,14 @@ buildPythonPackage (finalAttrs: { 
    hash = "sha256-Vjv62cYDIuTLE7MxRt4Havy7DMOiMTyIixbs4LGFGGs=";

  };



  # TypeError: NDArray.record() missing 1 required keyword-only argument: 'in_warmup'

  postPatch = ''

    substituteInPlace bambi/backend/pymc.py \

      --replace-fail \

        "strace.record(point=dict(zip(varnames, value)))" \

        "strace.record(point=dict(zip(varnames, value)), in_warmup=False)"

  '';



  build-system = [

    setuptools

    setuptools-scm

pkgs/development/python-modules/blackjax/default.nix

+6 −4
Original line number Diff line number Diff line
@@ -15,11 +15,12 @@ 
  typing-extensions,



  # checks

  chex,

  pytestCheckHook,

  pytest-xdist,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "blackjax";

  version = "1.3";

  pyproject = true;
@@ -27,7 +28,7 @@ buildPythonPackage rec { 
  src = fetchFromGitHub {

    owner = "blackjax-devs";

    repo = "blackjax";

    tag = version;

    tag = finalAttrs.version;

    hash = "sha256-ystvPfIsnMFYkC+LNtcRQsI19i/y/905SnPSApM8v4E=";

  };
@@ -46,6 +47,7 @@ buildPythonPackage rec { 
  ];



  nativeCheckInputs = [

    chex

    pytestCheckHook

    pytest-xdist

  ];
@@ -83,8 +85,8 @@ buildPythonPackage rec { 
  meta = {

    homepage = "https://blackjax-devs.github.io/blackjax";

    description = "Sampling library designed for ease of use, speed and modularity";

    changelog = "https://github.com/blackjax-devs/blackjax/releases/tag/${version}";

    changelog = "https://github.com/blackjax-devs/blackjax/releases/tag/${finalAttrs.src.tag}";

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ bcdarwin ];

  };

}
 
})

pkgs/development/python-modules/brax/default.nix

+11 −8
Original line number Diff line number Diff line
@@ -36,16 +36,16 @@ 
  transforms3d,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "brax";

  version = "0.14.0";

  version = "0.14.2";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "google";

    repo = "brax";

    tag = "v${version}";

    hash = "sha256-CkRXEYtlP8IhEZ7lVnpxlwiyLdICAfILwHfRUfuub08=";

    tag = "v${finalAttrs.version}";

    hash = "sha256-/oznBa44xKl+9T1YrTVhCbuKZj16V1BTlnmCGRbF45g=";

  };



  build-system = [
@@ -88,6 +88,11 @@ buildPythonPackage rec { 
    "testModelEncoding1"

    "testTrain"

    "testTrainDomainRandomize"



    # ValueError: Error: no decoder found for mesh file 'meshes/pyramid.stl'

    "test_convex_convex"

    "test_dumps"

    "test_dumps_invalidstate_raises"

  ]

  ++ lib.optionals stdenv.hostPlatform.isAarch64 [

    # Flaky:
@@ -100,9 +105,7 @@ buildPythonPackage rec { 
    "brax/generalized/constraint_test.py"

  ];



  pythonImportsCheck = [

    "brax"

  ];

  pythonImportsCheck = [ "brax" ];



  meta = {

    description = "Massively parallel rigidbody physics simulation on accelerator hardware";
@@ -110,4 +113,4 @@ buildPythonPackage rec { 
    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ nim65s ];

  };

}
 
})

pkgs/development/python-modules/clu/default.nix

+6 −4
Original line number Diff line number Diff line
@@ -19,6 +19,7 @@ 
  wrapt,



  # tests

  chex,

  keras,

  pytestCheckHook,

  tensorflow,
@@ -26,7 +27,7 @@ 
  torch,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "clu";

  version = "0.0.12";

  pyproject = true;
@@ -34,7 +35,7 @@ buildPythonPackage rec { 
  src = fetchFromGitHub {

    owner = "google";

    repo = "CommonLoopUtils";

    tag = "v${version}";

    tag = "v${finalAttrs.version}";

    hash = "sha256-ntqRz3fCXMf0EDQsddT68Mdi105ECBVQpVsStzk2kvQ=";

  };
@@ -59,6 +60,7 @@ buildPythonPackage rec { 
  pythonImportsCheck = [ "clu" ];



  nativeCheckInputs = [

    chex

    keras

    pytestCheckHook

    tensorflow
@@ -76,8 +78,8 @@ buildPythonPackage rec { 
  meta = {

    description = "Common training loops in JAX";

    homepage = "https://github.com/google/CommonLoopUtils";

    changelog = "https://github.com/google/CommonLoopUtils/blob/${src.tag}/CHANGELOG.md";

    changelog = "https://github.com/google/CommonLoopUtils/blob/${finalAttrs.src.tag}/CHANGELOG.md";

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ GaetanLepage ];

  };

}
 
})

pkgs/development/python-modules/distrax/default.nix

+7 −4
Original line number Diff line number Diff line
@@ -20,7 +20,7 @@ 
  pytestCheckHook,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "distrax";

  version = "0.1.7";

  pyproject = true;
@@ -28,7 +28,7 @@ buildPythonPackage rec { 
  src = fetchFromGitHub {

    owner = "google-deepmind";

    repo = "distrax";

    tag = "v${version}";

    tag = "v${finalAttrs.version}";

    hash = "sha256-R6rGGNzup3O6eZ2z4vygYWTjroE/Irt3aog8Op+0hco=";

  };
@@ -57,6 +57,9 @@ buildPythonPackage rec { 
  pythonImportsCheck = [ "distrax" ];



  disabledTests = [

    # execnet.gateway_base.DumpError: can't serialize <class 'method'>

    "test_raises_on_invalid_input_shape"



    # Flaky: AssertionError: 1 not less than 0.7000000000000001

    "test_von_mises_sample_uniform_ks_test"
@@ -124,7 +127,7 @@ buildPythonPackage rec { 
  meta = {

    description = "Probability distributions in JAX";

    homepage = "https://github.com/deepmind/distrax";

    changelog = "https://github.com/google-deepmind/distrax/releases/tag/v${version}";

    changelog = "https://github.com/google-deepmind/distrax/releases/tag/${finalAttrs.src.tag}";

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ onny ];

    badPlatforms = [
@@ -132,4 +135,4 @@ buildPythonPackage rec { 
      lib.systems.inspect.patterns.isDarwin

    ];

  };

}
 
})