gnu: Add python-einops.

* gnu/packages/python-science.scm (python-einops): New variable.
This commit is contained in:
Vinicius Monego 2023-05-20 15:24:19 -03:00
parent fa214b27cf
commit 6bc2fbc14c
No known key found for this signature in database
GPG key ID: 637B0B138065B68A

View file

@ -669,6 +669,40 @@ (define-public python-geosketch
necessary for reproducing the experiments in the paper.")
(license license:expat)))
(define-public python-einops
(package
(name "python-einops")
(version "0.6.1")
(source (origin
(method git-fetch) ;PyPI misses .ipynb files required for tests
(uri (git-reference
(url "https://github.com/arogozhnikov/einops")
(commit (string-append "v" version))))
(file-name (git-file-name name version))
(sha256
(base32
"1h8p39kd7ylg99mh620xr20hg7v78x1jnj6vxwk31rlw2dmv2dpr"))))
(build-system pyproject-build-system)
(arguments
(list #:phases #~(modify-phases %standard-phases
(add-after 'unpack 'set-backend
(lambda _
;; Einops supports different backends, but we test
;; only NumPy for availability and simplicity.
(setenv "EINOPS_TEST_BACKENDS" "numpy"))))))
(native-inputs (list jupyter
python-hatchling
python-nbconvert
python-nbformat
python-parameterized
python-pytest))
(propagated-inputs (list python-numpy))
(home-page "https://einops.rocks/")
(synopsis "Tensor operations for different backends")
(description "Einops provides a set of tensor operations for NumPy and
multiple deep learning frameworks.")
(license license:expat)))
(define-public python-xarray
(package
(name "python-xarray")