WS-KCNHUB/machines/kcnhub/packages/python.nix

70 lines
1.5 KiB
Nix
Raw Normal View History

2023-08-17 10:37:59 -04:00
{config, lib, pkgs, ...}: {
options = {
pythonPackages = lib.mkOption {
type = lib.types.anything;
default = p: with p; [];
example = p: with p; [ numpy scipy ];
description = "List of Python Packages installed";
};
};
config = {
environment.systemPackages = with pkgs; [
((pkgs.python3.withPackages config.pythonPackages).override (args: { ignoreCollisions = true; }))
];
pythonPackages = (p: with p; [
# Common Python Libraries
2023-08-17 10:37:59 -04:00
numpy
scipy
scikit-learn
2023-08-17 10:37:59 -04:00
pandas
2024-05-14 11:44:56 -04:00
openpyxl
2023-08-17 10:37:59 -04:00
matplotlib
ipympl
matplotlib-inline
2023-08-17 10:37:59 -04:00
seaborn
# Neural Data Processing Libraries
# Temporarily broken?: https://github.com/NixOS/nixpkgs/issues/259812
mne-python
2024-05-14 11:44:56 -04:00
edfio
nibabel
# View niftis!
ipyniivue
neo
# Neurosimulators
neuronpy
# nest with MPI is incompatible with Jupyter:
# https://www.nest-simulator.org/pynest-api/_modules/nest.html
# see pynestkernel comments about their workaround
(nest.override {
withMpi = false;
})
brian2
2024-07-31 15:35:41 -04:00
# SpiNNaker simulator
spynnaker
2024-11-16 10:56:36 -05:00
# fetching neuromorphic data:
tonic
# Machine learning toolkits:
tensorflow
tensorboard
keras
edward
transformers
tflearn
torch
torchvision
torchsde
torchmetrics
torchio
torchdiffeq
botorch
lion-pytorch
pycuda
accelerate
datasets
]);
};
2023-08-17 10:37:59 -04:00
}