python3Packages.pymc: 5.26.1 -> 5.27.0 (#474390)

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

@@ -91,9 +91,10 @@ buildPythonPackage rec {
     "test_model_without_intercept"
     "test_non_distributional_model"
     "test_normal_with_splines"
-    "test_predict_new_groups_fail"
     "test_predict_new_groups"
+    "test_predict_new_groups_fail"
     "test_predict_offset"
+    "test_same_variable_conditional_and_group"
     "test_set_alias_warnings"
     "test_subplot_kwargs"
     "test_transforms"
@@ -113,8 +114,6 @@ buildPythonPackage rec {
   ];
 
   disabledTestPaths = [
-    # bayeux-ml is not available
-    "tests/test_alternative_samplers.py"
     # Tests require network access
     "tests/test_interpret.py"
     "tests/test_interpret_messages.py"

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

@@ -18,18 +18,20 @@
   scipy,
   threadpoolctl,
   typing-extensions,
+
+  writableTmpDirAsHomeHook,
 }:
 
 buildPythonPackage rec {
   pname = "pymc";
-  version = "5.26.1";
+  version = "5.27.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "pymc-devs";
     repo = "pymc";
     tag = "v${version}";
-    hash = "sha256-j1v8MzAFfOmkN7pDcF91dS5Xprls8qfTQHWdaFUO4GU=";
+    hash = "sha256-wBeWydrHrF+wNZnqWa2k8tCaUvjcoiSrmY85LUhrQds=";
   };
 
   build-system = [
@@ -50,6 +52,13 @@ buildPythonPackage rec {
     typing-extensions
   ];
 
+  nativeBuildInputs = [
+    # Arviz (imported by pymc) wants to write a stamp file to the homedir at import time.
+    # Without $HOME being writable, `pythonImportsCheck` fails.
+    # https://github.com/arviz-devs/arviz/commit/4db612908f588d89bb5bfb6b83a08ada3d54fd02
+    writableTmpDirAsHomeHook
+  ];
+
   # The test suite is computationally intensive and test failures are not
   # indicative for package usability hence tests are disabled by default.
   doCheck = false;

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

@@ -3,6 +3,7 @@
   stdenv,
   buildPythonPackage,
   fetchFromGitHub,
+  fetchpatch,
 
   # build-system
   setuptools,
@@ -33,7 +34,7 @@
 
 buildPythonPackage rec {
   pname = "pytensor";
-  version = "2.36.0";
+  version = "2.36.1";
   pyproject = true;
 
   src = fetchFromGitHub {
@@ -43,9 +44,18 @@ buildPythonPackage rec {
     postFetch = ''
       sed -i 's/git_refnames = "[^"]*"/git_refnames = " (tag: ${src.tag})"/' $out/pytensor/_version.py
     '';
-    hash = "sha256-tzwiPp0+xNKmndTn9Y1AXiqscQWaCC8gKgQHEtkyGag=";
+    hash = "sha256-rXLtrkuwmEe5+64Aao490VqD96LJ37/mxekWOzWRMlw=";
   };
 
+  patches = [
+    # https://github.com/pymc-devs/pytensor/pull/1805
+    (fetchpatch {
+      name = "fix-test-tri-nonconcrete-jax-compatibility.patch";
+      url = "https://github.com/pymc-devs/pytensor/commit/86310f074267e24d1b3b99ecf3d9cc0b593b170d.patch";
+      hash = "sha256-KRywJLixmdDJ1GGYsd5Twjiwgce0ZFxUidhTgM6Obmg=";
+    })
+  ];
+
   build-system = [
     setuptools
     cython
@@ -160,7 +170,6 @@ buildPythonPackage rec {
     # Don't run the most compute-intense tests
     "tests/scan/"
     "tests/tensor/"
-    "tests/sparse/sandbox/"
   ];
 
   passthru.updateScript = nix-update-script {