Commit b892aeec authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python312Packages.distrax: disable failing tests

parent 93dbaf6d

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

+11 −7
Original line number Diff line number Diff line 
{

  lib,

  buildPythonPackage,

  pythonOlder,

  fetchFromGitHub,

  chex,

  jaxlib,
@@ -17,8 +16,6 @@ buildPythonPackage rec { 
  version = "0.1.5";

  pyproject = true;



  disabled = pythonOlder "3.9";



  src = fetchFromGitHub {

    owner = "google-deepmind";

    repo = "distrax";
@@ -26,7 +23,7 @@ buildPythonPackage rec { 
    hash = "sha256-A1aCL/I89Blg9sNmIWQru4QJteUTN6+bhgrEJPmCrM0=";

  };



  buildInputs = [

  dependencies = [

    chex

    jaxlib

    numpy
@@ -42,6 +39,12 @@ buildPythonPackage rec { 
  pythonImportsCheck = [ "distrax" ];



  disabledTests = [

    # NotImplementedError: Primitive 'square' does not have a registered inverse.

    "test_against_tfp_bijectors_square"

    "test_log_dets_square__with_device"

    "test_log_dets_square__without_device"

    "test_log_dets_square__without_jit"



    # AssertionError on numerical values

    # Reported upstream in https://github.com/google-deepmind/distrax/issues/267

    "test_method_with_input_unnormalized_probs__with_device"
@@ -90,10 +93,11 @@ buildPythonPackage rec { 
    "distrax/_src/utils/hmm_test.py"

  ];



  meta = with lib; {

  meta = {

    description = "Probability distributions in JAX";

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

    license = licenses.asl20;

    maintainers = with maintainers; [ onny ];

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

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ onny ];

  };

}