Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.8 kB · 97 lines
Nix
at litex
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798{ stdenv, lib, buildPythonPackage, cons, cython, etuples, fetchFromGitHub, filelock, jax, jaxlib, logical-unification, minikanren, numba, numba-scipy, numpy, pytestCheckHook, pythonOlder, scipy, typing-extensions}:
buildPythonPackage rec { pname = "pytensor"; version = "2.11.3"; format = "setuptools";
disabled = pythonOlder "3.7";
src = fetchFromGitHub { owner = "pymc-devs"; repo = pname; rev = "refs/tags/rel-${version}"; hash = "sha256-4GDur8S19i8pZkywKHZUelmd2e0jZmC5HzF7o2esDl4="; };
nativeBuildInputs = [ cython ];
propagatedBuildInputs = [ cons etuples filelock logical-unification minikanren numpy scipy typing-extensions ];
checkInputs = [ jax jaxlib numba numba-scipy pytestCheckHook ];
postPatch = '' substituteInPlace setup.cfg \ --replace "--durations=50" "" '';
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/sandbox/" "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); };}