Merge pull request #255882 from GaetanLepage/pytensor (01214215) · Commits · nix / nixpkgs

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

+32 −21

Original line number	Diff line number	Diff line
		{ stdenv
		, lib
		{ lib
		, buildPythonPackage
		, cons
		, fetchFromGitHub
		, cython
		, versioneer
		, cons
		, etuples
		, fetchFromGitHub
		, filelock
		, jax
		, jaxlib
		, logical-unification
		, minikanren
		, numpy
		, scipy
		, typing-extensions
		, jax
		, jaxlib
		, numba
		, numba-scipy
		, numpy
		, pytest-mock
		, pytestCheckHook
		, pythonOlder
		, scipy
		, typing-extensions
		# Tensorflow is currently (2023/10/04) broken.
		# Thus, we don't provide this optional test dependency.
		# , tensorflow-probability
		, stdenv
		}:

		buildPythonPackage rec {
		pname = "pytensor";
		version = "2.11.3";
		format = "setuptools";
		version = "2.17.1";
		pyproject = true;

		disabled = pythonOlder "3.7";
		disabled = pythonOlder "3.9";

		src = fetchFromGitHub {
		owner = "pymc-devs";
		repo = pname;
		repo = "pytensor";
		rev = "refs/tags/rel-${version}";
		hash = "sha256-4GDur8S19i8pZkywKHZUelmd2e0jZmC5HzF7o2esDl4=";
		hash = "sha256-OG7ZvXJOTQzSpMoj94K9JwG6ZQRQSSp+k2slkK6BJ9I=";
		};

		postPatch = ''
		substituteInPlace pyproject.toml \
		--replace "versioneer[toml]==0.28" "versioneer[toml]"
		'';

		nativeBuildInputs = [
		cython
		versioneer
		];

		propagatedBuildInputs = [
		@@ -48,19 +59,18 @@ buildPythonPackage rec {
		typing-extensions
		];

		checkInputs = [
		nativeCheckInputs = [
		jax
		jaxlib
		numba
		numba-scipy
		pytest-mock
		pytestCheckHook
		# Tensorflow is currently (2023/10/04) broken.
		# Thus, we don't provide this optional test dependency.
		# tensorflow-probability
		];

		postPatch = ''
		substituteInPlace setup.cfg \
		--replace "--durations=50" ""
		'';

		preBuild = ''
		export HOME=$(mktemp -d)
		'';
		@@ -76,13 +86,14 @@ buildPythonPackage rec {
		"test_logsumexp_benchmark"
		"test_scan_multiple_output"
		"test_vector_taps_benchmark"
		# Temporarily disabled because of broken tensorflow-probability
		"test_tfp_ops"
		];

		disabledTestPaths = [
		# Don't run the most compute-intense tests
		"tests/scan/"
		"tests/tensor/"
		"tests/sandbox/"
		"tests/sparse/sandbox/"
		];