From fd802eebeff2b09408f4845308fac3fbca6ba0e3 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Thu, 6 Feb 2025 10:32:05 +0100
Subject: [PATCH] python312Packages.arviz: fix build

---
 .../development/python-modules/arviz/default.nix | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/pkgs/development/python-modules/arviz/default.nix b/pkgs/development/python-modules/arviz/default.nix
index 5388b5abfc1a..e7c98a11919d 100644
--- a/pkgs/development/python-modules/arviz/default.nix
+++ b/pkgs/development/python-modules/arviz/default.nix
@@ -18,7 +18,7 @@
   xarray,
   xarray-einstats,
 
-  # checks
+  # tests
   bokeh,
   cloudpickle,
   emcee,
@@ -33,6 +33,7 @@
   #, pystan (not packaged)
   pytestCheckHook,
   torchvision,
+  writableTmpDirAsHomeHook,
   zarr,
 }:
 
@@ -79,14 +80,18 @@ buildPythonPackage rec {
     # pystan (not packaged)
     pytestCheckHook
     torchvision
+    writableTmpDirAsHomeHook
     zarr
   ];
 
-  preCheck = ''
-    export HOME=$(mktemp -d);
-  '';
+  pytestFlagsArray = [
+    "arviz/tests/base_tests/"
 
-  pytestFlagsArray = [ "arviz/tests/base_tests/" ];
+    # AttributeError: module 'zarr.storage' has no attribute 'DirectoryStore'
+    # https://github.com/arviz-devs/arviz/issues/2357
+    "--deselect=arviz/tests/base_tests/test_data_zarr.py::TestDataZarr::test_io_function"
+    "--deselect=arviz/tests/base_tests/test_data_zarr.py::TestDataZarr::test_io_method"
+  ];
 
   disabledTests = [
     # Tests require network access
@@ -94,6 +99,7 @@ buildPythonPackage rec {
     "test_plot_separation"
     "test_plot_trace_legend"
     "test_cov"
+
     # countourpy is not available at the moment
     "test_plot_kde"
     "test_plot_kde_2d"