summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorNavid Afkhami <navid.afkhami@mdc-berlin.de>2023-05-25 17:18:46 +0200
committerRicardo Wurmus <rekado@elephly.net>2023-05-25 17:19:29 +0200
commit9f37dc3be3780f644efd4bd5af2ff039ebb51d75 (patch)
tree735424b6c2c280cc61f81eba19cfa8f5f24b55fd
parent99fc7e5dd3c9c92f9b6c2e51db9dc5b0481a56b2 (diff)
gnu: Add python-tensorly.
* gnu/packages/machine-learning.scm (python-tensorly): New variable. Co-authored-by: Ricardo Wurmus <rekado@elephly.net>
-rw-r--r--gnu/packages/machine-learning.scm41
1 files changed, 41 insertions, 0 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm
index 5fa30ebfff..462e8cf54b 100644
--- a/gnu/packages/machine-learning.scm
+++ b/gnu/packages/machine-learning.scm
@@ -19,6 +19,7 @@
;;; Copyright © 2020, 2021, 2022, 2023 Maxim Cournoyer <maxim.cournoyer@gmail.com>
;;; Copyright © 2022, 2023 Nicolas Graves <ngraves@ngraves.fr>
;;; Copyright © 2023 zamfofex <zamfofex@twdb.moe>
+;;; Copyright © 2023 Navid Afkhami <navid.afkhami@mdc-berlin.de>
;;;
;;; This file is part of GNU Guix.
;;;
@@ -3109,6 +3110,46 @@ These include a barrier, broadcast, and allreduce.")
(home-page "https://github.com/facebookincubator/gloo")
(license license:bsd-3))))
+(define-public python-tensorly
+ (package
+ (name "python-tensorly")
+ (version "0.8.1")
+ (source (origin
+ (method git-fetch)
+ (uri (git-reference
+ (url "https://github.com/tensorly/tensorly")
+ (commit version)))
+ (file-name (git-file-name name version))
+ (sha256
+ (base32
+ "184mvs1gwycsh2f8jgdyc3jyhiylbn4xw2srpjd264dz2l9ms2l7"))))
+ (build-system pyproject-build-system)
+ (arguments
+ (list
+ #:test-flags
+ '(list
+ ;; These tests fail due to missing example, documentation, or tutorial files.
+ "--ignore=doc/sphinx_ext/sphinx_gallery/tests/test_gen_rst.py"
+ ;; XXX There is no "get_marker" method.
+ "--ignore=doc/sphinx_ext/sphinx_gallery/tests/test_gen_gallery.py"
+ "-k"
+ (string-append
+ ;; tutorials/plot_parse.py is not included
+ "not test_jupyter_notebook"
+ ;; nor is examples/plot_quantum.py
+ " and not test_file_is_generated"))))
+ (propagated-inputs (list python-jsmin python-numpy python-scipy))
+ (native-inputs (list python-pytest python-pytest-cov python-sphinx))
+ (home-page "https://github.com/tensorly/tensorly")
+ (synopsis "Tensor learning in Python")
+ (description
+ "This is a Python library that aims at making tensor learning simple and
+accessible. It allows performing tensor decomposition, tensor learning and
+tensor algebra easily. Its backend system allows seamlessly perform
+computation with NumPy, PyTorch, JAX, MXNet, TensorFlow or CuPy and run
+methodxs at scale on CPU or GPU.")
+ (license license:bsd-3)))
+
(define-public python-umap-learn
(package
(name "python-umap-learn")