manifest.scm

;; Use this to set up a development environment.
;;   guix time-machine -C channels.scm -- shell -m manifest.scm
;;
;; Optionally, add "--pure" or "--container" for more isolation.

(import (guix packages)
        (guix profiles))

(define replace-cpu-torch-with-gpu-torch
  (if (getenv "FLEXYNESIS_USE_CPU")
      identity  ;don't do anything
      (let ((cpu-torch (specification->package "python-pytorch"))
            (gpu-torch (specification->package "python-pytorch-with-cuda11")))
        (package-input-rewriting `((,cpu-torch . ,gpu-torch))))))

(define %packages
  (map replace-cpu-torch-with-gpu-torch
       (map specification->package
            (list "python-captum"
                  "python-ipywidgets"
                  "python-louvain"
                  "python-lifelines"
                  "python-matplotlib"
                  "python-numpy"
                  "python-pandas"
                  "python-papermill"
                  "python-pytorch"
                  "python-pytorch-lightning"
                  "python-pytorch-geometric"
                  "python-pyyaml"
                  "python-rich"
                  "python-scikit-optimize"
                  "python-scikit-survival"
                  "python-scipy"
                  "python-seaborn"
                  "python-torchvision"
                  "python-tqdm"
                  "python-umap-learn"))))
(define %dev-packages
  (map specification->package
       (list "python-pytest")))

(packages->manifest
 (cons (specification->package "python")
       (append %packages %dev-packages)))