From 94d5b82ce8cf684a6eb72562243b7fc96dd9892c Mon Sep 17 00:00:00 2001
From: Ricardo Wurmus <rekado@elephly.net>
Date: Wed, 25 Oct 2023 11:10:51 +0200
Subject: [PATCH] python: Add python-tensorflow.

* guix-science/packages/python.scm (python-tensorflow): New variable.
---
 guix-science/packages/python.scm | 84 ++++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)

diff --git a/guix-science/packages/python.scm b/guix-science/packages/python.scm
index c93f475..c9ec380 100644
--- a/guix-science/packages/python.scm
+++ b/guix-science/packages/python.scm
@@ -26,11 +26,13 @@
   #:use-module (gnu packages check)
   #:use-module (gnu packages compression)
   #:use-module (gnu packages curl)
+  #:use-module (gnu packages elf)
   #:use-module (gnu packages graph)
   #:use-module (gnu packages icu4c)
   #:use-module (gnu packages image)
   #:use-module (gnu packages java)
   #:use-module (gnu packages linux)
+  #:use-module (gnu packages llvm)
   #:use-module (gnu packages maths)
   #:use-module (gnu packages mpi)
   #:use-module (gnu packages pdf)
@@ -1695,3 +1697,85 @@ including both a sequential API for beginners that allows users to
 build models quickly by plugging together building blocks and a
 subclassing API with an imperative style for advanced research.")
     (license license:asl2.0)))
+
+(define-public python-tensorflow
+  (package
+    (inherit tensorflow)
+    (name "python-tensorflow")
+    (source #f)
+    (build-system pyproject-build-system)
+    (arguments
+     (list
+      #:tests? #false
+      #:phases
+      #~(modify-phases %standard-phases
+          (replace 'unpack
+            (lambda _
+              (mkdir-p "source")
+              (copy-recursively #$tensorflow:python "source")
+              (chdir "source")
+              ;; XXX: the "python" output of the tensorflow package
+              ;; contains broken symlinks.
+              (delete-file-recursively "third_party/eigen3")
+              (mkdir-p "third_party/eigen3")
+              (copy-recursively #$eigen-for-python-ml-dtypes
+                                "third_party/eigen3")
+              (with-output-to-file "third_party/eigen3/LICENSE"
+                (lambda () (display "")))))
+          (add-after 'unpack 'relax-dependencies
+            (lambda _
+              (substitute* "setup.py"
+                ;; We don't have tensorflow-io yet
+                (("'tensorflow-io-gcs-filesystem.*") "None,")
+                (("'platform_system.*") "")
+                ;; Versions above 0.4 break tests, but that's okay
+                ;; because we aren't running them.
+                (("gast >= 0.2.1, <= 0.4.0") "gast >= 0.2.1")
+                ;; Drop all of tensorboard, keras, and tensorflow_estimator
+                (("'(keras|tensorboard|tensorflow_estimator) >.*',") " None,")
+                ;; Our clang bindings have a different name.
+                (("libclang") "clang")
+                ;; No tensorboard, sorry.
+                (("standard_or_nightly\\('tensorboard = tensorboard.main:run_main', None\\),") "")
+                (("'import_pb_to_tensorboard = tensorflow.python.tools.import_pb_to_tensorboard:main',") "")
+                ;; We don't have tensorflow-estimator yet.
+                (("'estimator_ckpt_converter = '") "")
+                (("'tensorflow_estimator.python.estimator.tools.checkpoint_converter:main',") ""))))
+          ;; XXX: this is really ugly, but many shared objects cannot
+          ;; find libtensorflow_framework.so.2 and libbfloat16.so.so
+          (add-after 'unpack 'find-tensorflow-libraries
+            (lambda _
+              ;; XXX: not all .so files need this.
+              (let ((libraries (find-files "." ".*\\.so$")))
+                (for-each (lambda (lib)
+                            (make-file-writable lib)
+                            (system (format #false
+                                            "patchelf --set-rpath ~a:~a:$(patchelf --print-rpath ~a) ~a"
+                                            ;; for libtensorflow_framework.so.2
+                                            (string-append
+                                             #$(this-package-input "tensorflow")
+                                             "/lib")
+                                            ;; for libbfloat16.so.so
+                                            (string-append
+                                             #$output
+                                             "/lib/python3.10/site-packages/tensorflow/tsl/python/lib/core/")
+                                            lib lib)))
+                          libraries))))
+          (add-after 'install 'install-missing-libraries
+            (lambda _
+              ;; libtensorflow_cc.so.2 is not installed.  See
+              ;; https://github.com/tensorflow/tensorflow/issues/60326.
+              (let ((dir (string-append #$output "/lib/python3.10/site-packages/tensorflow/"))
+                    (lib (string-append "libtensorflow_cc.so." #$(package-version this-package))))
+                (install-file (string-append "tensorflow/" lib) dir)
+                (with-directory-excursion dir
+                  (symlink lib "libtensorflow_cc.so.2"))))))))
+    (outputs '("out"))
+    (propagated-inputs
+     (modify-inputs (package-propagated-inputs tensorflow)
+       (append python-clang-13)))
+    (inputs (list tensorflow))
+    (native-inputs
+     (list eigen-for-python-ml-dtypes
+           patchelf
+           `(,tensorflow "python")))))