Unverified Commit b11ced7a authored by Mario Rodas's avatar Mario Rodas Committed by GitHub
Browse files

Merge pull request #260359 from bcdarwin/update-jaxopt 

python310Packages.jaxopt: 0.5.5 -> 0.8.1
parents 484cab6a c99a99ce

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

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


buildPythonPackage rec {

  pname = "bambi";

  version = "0.10.0";

  format = "pyproject";

  version = "0.12.0";

  pyproject = true;



  disabled = pythonOlder "3.8";

  disabled = pythonOlder "3.9";



  src = fetchFromGitHub {

    owner = "bambinos";

    repo = pname;

    repo = "bambi";

    rev = "refs/tags/${version}";

    hash = "sha256-D04eTAlckEqgKA+59BRljlyneHYoqqZvLYmt/gBLHcU=";

    hash = "sha256-36D8u813v2vWQdNqBWfM8YVnAJuLGvn5vqdHs94odmU=";

  };



  nativeBuildInputs = [
@@ -55,16 +55,18 @@ buildPythonPackage rec { 


  disabledTests = [

    # Tests require network access

    "test_alias_equal_to_name"

    "test_custom_prior"

    "test_data_is_copied"

    "test_distributional_model"

    "test_extra_namespace"

    "test_gamma_with_splines"

    "test_non_distributional_model_with_categories"

    "test_non_distributional_model"

    "test_normal_with_splines"

    "test_predict_offset"

    # Assertion issue

    "test_custom_likelihood_function"

    "test_predict_new_groups"

    "test_predict_new_groups_fail"

    "test_set_alias_warnings"

  ];



  pythonImportsCheck = [
@@ -74,6 +76,7 @@ buildPythonPackage rec { 
  meta = with lib; {

    homepage = "https://bambinos.github.io/bambi";

    description = "High-level Bayesian model-building interface";

    changelog = "https://github.com/bambinos/bambi/releases/tag/${version}";

    license = licenses.mit;

    maintainers = with maintainers; [ bcdarwin ];

  };

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

+10 −12
Original line number Diff line number Diff line
@@ -2,8 +2,8 @@ 
, buildPythonPackage

, pythonOlder

, fetchFromGitHub

, fetchpatch

, pytestCheckHook

, setuptools-scm

, fastprogress

, jax

, jaxlib
@@ -14,24 +14,21 @@ 


buildPythonPackage rec {

  pname = "blackjax";

  version = "0.9.6";

  disabled = pythonOlder "3.7";

  version = "1.0.0";

  pyproject = true;



  disabled = pythonOlder "3.8";



  src = fetchFromGitHub {

    owner = "blackjax-devs";

    repo = pname;

    rev = "refs/tags/${version}";

    hash = "sha256-EieDu9SJxi2cp1bHlxX4vvFZeDGMGIm24GoR8nSyjvE=";

    hash = "sha256-hqOKSHyZ/BmOu6MJLeecD3H1BbLbZqywmlBzn3xjQRk=";

  };



  patches = [

    # remove in next release

    (fetchpatch {

      name = "fix-lbfgs-args";

      url = "https://github.com/blackjax-devs/blackjax/commit/1aaa6f64bbcb0557b658604b2daba826e260cbc6.patch";

      hash = "sha256-XyjorXPH5Ap35Tv1/lTeTWamjplJF29SsvOq59ypftE=";

    })

  ];

  nativeBuildInputs = [ setuptools-scm ];



  env.SETUPTOOLS_SCM_PRETEND_VERSION = version;



  propagatedBuildInputs = [

    fastprogress
@@ -56,6 +53,7 @@ buildPythonPackage rec { 
  meta = with lib; {

    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}";

    license = licenses.asl20;

    maintainers = with maintainers; [ bcdarwin ];

  };

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

+3 −2
Original line number Diff line number Diff line
@@ -15,7 +15,7 @@ 


buildPythonPackage rec {

  pname = "chex";

  version = "0.1.82";

  version = "0.1.83";

  format = "setuptools";



  disabled = pythonOlder "3.9";
@@ -24,7 +24,7 @@ buildPythonPackage rec { 
    owner = "deepmind";

    repo = pname;

    rev = "refs/tags/v${version}";

    hash = "sha256-xBq22AaR2Tp1NSPefEyvCDeUYqRZlAf5LVHWo0luiXk=";

    hash = "sha256-iEachJf5NjOnkMWdP0aVQHWNPgUUBkMnzHKq3GP7t4w=";

  };



  propagatedBuildInputs = [
@@ -49,6 +49,7 @@ buildPythonPackage rec { 
  meta = with lib; {

    description = "Chex is a library of utilities for helping to write reliable JAX code.";

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

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

    license = licenses.asl20;

    maintainers = with maintainers; [ ndl ];

  };

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

+4 −3
Original line number Diff line number Diff line
@@ -16,16 +16,16 @@ 


buildPythonPackage rec {

  pname = "jaxopt";

  version = "0.5.5";

  version = "0.8.1";

  format = "setuptools";



  disabled = pythonOlder "3.7";

  disabled = pythonOlder "3.8";



  src = fetchFromGitHub {

    owner = "google";

    repo = pname;

    rev = "refs/tags/${pname}-v${version}";

    hash = "sha256-WOsr/Dvguu9/qX6+LMlAKM3EANtYPtDu8Uo2157+bs0=";

    hash = "sha256-5+GfRFzXyc1Ukp86NeK0qstTq9fkx+eDb4iXvxzRoKs=";

  };



  propagatedBuildInputs = [
@@ -60,6 +60,7 @@ buildPythonPackage rec { 
  meta = with lib; {

    homepage = "https://jaxopt.github.io";

    description = "Hardware accelerated, batchable and differentiable optimizers in JAX";

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

    license = licenses.asl20;

    maintainers = with maintainers; [ bcdarwin ];

  };