1 files changed, 46 insertions, 55 deletions
diff --git a/pkgs/development/python-modules/arviz/default.nix b/pkgs/development/python-modules/arviz/default.nix
index b7860aa5dc30..0406894c8b05 100644
--- a/pkgs/development/python-modules/arviz/default.nix
+++ b/pkgs/development/python-modules/arviz/default.nix
@@ -1,48 +1,51 @@
-{ lib
-, buildPythonPackage
-, dm-tree
-, fetchFromGitHub
-, emcee
-, h5netcdf
-, matplotlib
-, netcdf4
-, numba
-, numpy
-, pandas
-, setuptools
-, cloudpickle
-, pytestCheckHook
-, scipy
-, packaging
-, pythonOlder
-, typing-extensions
-, xarray
-, xarray-einstats
-, zarr
-, ffmpeg
-, h5py
-, jaxlib
-, torchvision
-, jax
-  # , pymc3 (circular dependency)
-, pyro-ppl
+{
+  lib,
+  stdenv,
+  buildPythonPackage,
+  fetchFromGitHub,
+
+  # build-system
+  packaging,
+  setuptools,
+
+  # dependencies
+  h5netcdf,
+  matplotlib,
+  numpy,
+  pandas,
+  scipy,
+  typing-extensions,
+  xarray,
+  xarray-einstats,
+
+  # checks
+  bokeh,
+  cloudpickle,
+  emcee,
+  ffmpeg,
+  h5py,
+  jax,
+  jaxlib,
+  numba,
+  numpyro,
+  #, pymc3 (circular dependency)
+  pyro-ppl,
   #, pystan (not packaged)
-, numpyro
-, bokeh
+  pytestCheckHook,
+  torchvision,
+  zarr,
 }:
 
 buildPythonPackage rec {
   pname = "arviz";
-  version = "0.18.0";
+  version = "0.20.0";
   pyproject = true;
 
-  disabled = pythonOlder "3.10";
-
   src = fetchFromGitHub {
     owner = "arviz-devs";
     repo = "arviz";
     rev = "refs/tags/v${version}";
-    hash = "sha256-SZRqSqChQBSA9/jBXN2ds9hh6TI3qZksHai1j2oVsq0=";
+    hash = "sha256-6toqOGwk8YbatfiDCTEG4r0z3zZAA8zcNVZJqqssYrY=";
   };
 
   build-system = [
@@ -51,10 +54,8 @@ buildPythonPackage rec {
   ];
 
   dependencies = [
-    dm-tree
     h5netcdf
     matplotlib
-    netcdf4
     numpy
     pandas
     scipy
@@ -64,6 +65,7 @@ buildPythonPackage rec {
   ];
 
   nativeCheckInputs = [
+    bokeh
     cloudpickle
     emcee
     ffmpeg
@@ -78,16 +80,13 @@ buildPythonPackage rec {
     pytestCheckHook
     torchvision
     zarr
-    bokeh
   ];
 
   preCheck = ''
     export HOME=$(mktemp -d);
   '';
 
-  pytestFlagsArray = [
-    "arviz/tests/base_tests/"
-  ];
+  pytestFlagsArray = [ "arviz/tests/base_tests/" ];
 
   disabledTests = [
     # Tests require network access
@@ -99,26 +98,18 @@ buildPythonPackage rec {
     "test_plot_kde"
     "test_plot_kde_2d"
     "test_plot_pair"
-    # Array mismatch
-    "test_plot_ts"
-    # The following two tests fail in a common venv-based setup.
-    # An issue has been opened upstream: https://github.com/arviz-devs/arviz/issues/2282
-    "test_plot_ppc_discrete"
-    "test_plot_ppc_discrete_save_animation"
-    # Assertion error
-    "test_data_zarr"
-    "test_plot_forest"
   ];
 
-  pythonImportsCheck = [
-    "arviz"
-  ];
+  # Tests segfault on darwin
+  doCheck = !stdenv.isDarwin;
+
+  pythonImportsCheck = [ "arviz" ];
 
-  meta = with lib; {
+  meta = {
     description = "Library for exploratory analysis of Bayesian models";
     homepage = "https://arviz-devs.github.io/arviz/";
     changelog = "https://github.com/arviz-devs/arviz/blob/v${version}/CHANGELOG.md";
-    license = licenses.asl20;
-    maintainers = with maintainers; [ omnipotententity ];
+    license = lib.licenses.asl20;
+    maintainers = with lib.maintainers; [ omnipotententity ];
   };
 }