Commit 06489209 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent 4cfb8dc9
Loading
Loading
Loading
Loading
+12 −3
Original line number Diff line number Diff line
@@ -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 {