Commit ed80bc68 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent ac3f47bb
Loading
Loading
Loading
Loading
+12 −8
Original line number Diff line number Diff line
@@ -31,19 +31,19 @@
  nix-update-script,
}:

buildPythonPackage rec {
buildPythonPackage (finalAttrs: {
  pname = "pytensor";
  version = "2.36.3";
  version = "2.37.0";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "pymc-devs";
    repo = "pytensor";
    tag = "rel-${version}";
    tag = "rel-${finalAttrs.version}";
    postFetch = ''
      sed -i 's/git_refnames = "[^"]*"/git_refnames = " (tag: ${src.tag})"/' $out/pytensor/_version.py
      sed -i 's/git_refnames = "[^"]*"/git_refnames = " (tag: ${finalAttrs.src.tag})"/' $out/pytensor/_version.py
    '';
    hash = "sha256-FiqNE97tZw8rHF6VAQScGoiibMI+7KHzAy3tzmZBya4=";
    hash = "sha256-N6TYK/qMux/a0ktQyCGYHZg3g8S6To8g1FXnILk9HIw=";
  };

  build-system = [
@@ -84,7 +84,11 @@ buildPythonPackage rec {
    rm -rf pytensor
  '';

  disabledTests = lib.optionals stdenv.hostPlatform.isDarwin [
  disabledTests = [
    # TypeError: jax_funcified_fgraph() takes 2 positional arguments but 3 were given
    "test_jax_Reshape_shape_graph_input"
  ]
  ++ lib.optionals stdenv.hostPlatform.isDarwin [
    # Numerical assertion error
    # tests.unittest_tools.WrongValue: WrongValue
    "test_op_sd"
@@ -174,10 +178,10 @@ buildPythonPackage rec {
    description = "Python library to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays";
    mainProgram = "pytensor-cache";
    homepage = "https://github.com/pymc-devs/pytensor";
    changelog = "https://github.com/pymc-devs/pytensor/releases/tag/${src.tag}";
    changelog = "https://github.com/pymc-devs/pytensor/releases/tag/${finalAttrs.src.tag}";
    license = lib.licenses.bsd3;
    maintainers = with lib.maintainers; [
      bcdarwin
    ];
  };
}
})