Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
2.0 kB · 102 lines
Nix
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103{ lib, buildPythonPackage, fetchFromGitHub, cython, versioneer, cons, etuples, filelock, logical-unification, minikanren, numpy, scipy, typing-extensions, jax, jaxlib, numba, numba-scipy, pytest-mock, pytestCheckHook, pythonOlder, tensorflow-probability, stdenv}:
buildPythonPackage rec { pname = "pytensor"; version = "2.18.6"; pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub { owner = "pymc-devs"; repo = "pytensor"; rev = "refs/tags/rel-${version}"; hash = "sha256-SMh4wVZwmc87ztFn2OOI234VP3JzmxVMBkn7lYwVu6M="; };
postPatch = '' substituteInPlace pyproject.toml \ --replace "versioneer[toml]==0.28" "versioneer[toml]" '';
nativeBuildInputs = [ cython versioneer ];
propagatedBuildInputs = [ cons etuples filelock logical-unification minikanren numpy scipy typing-extensions ];
nativeCheckInputs = [ jax jaxlib numba numba-scipy pytest-mock pytestCheckHook tensorflow-probability ];
preBuild = '' export HOME=$(mktemp -d) '';
pythonImportsCheck = [ "pytensor" ];
disabledTests = [ # benchmarks (require pytest-benchmark): "test_elemwise_speed" "test_fused_elemwise_benchmark" "test_logsumexp_benchmark" "test_scan_multiple_output" "test_vector_taps_benchmark" ];
disabledTestPaths = [ # Don't run the most compute-intense tests "tests/scan/" "tests/tensor/" "tests/sparse/sandbox/" ];
meta = with lib; { description = "Python library to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays"; homepage = "https://github.com/pymc-devs/pytensor"; changelog = "https://github.com/pymc-devs/pytensor/releases"; license = licenses.bsd3; maintainers = with maintainers; [ bcdarwin ]; broken = (stdenv.isLinux && stdenv.isAarch64); };}