File size: 1,430 Bytes
1c6540f
 
 
 
 
 
 
 
 
5fd2416
1c6540f
 
 
 
 
 
 
 
 
 
 
 
5fd2416
 
 
 
 
 
1c6540f
 
 
 
 
 
5fd2416
 
 
 
 
 
1c6540f
 
 
 
 
 
 
 
 
 
12d7151
 
1c6540f
5fd2416
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[tool.setuptools]
packages = ["shimnet"]   # only include this package

[project]
name = "shimnet"
version = "0.1.1"
description = "Package for ShimNet: A Neural Network for Postacquisition Improvement of NMR Spectra Distorted by Magnetic-Field Inhomogeneity https://pubs.acs.org/doi/full/10.1021/acs.jpcb.5c02632"
authors = [
    {name = "Marek Bukowicki", email = "m.bukowicki@uw.edu.pl"},
]
readme = "Readme.md"
requires-python = ">=3.10"
dependencies = [
]

[project.optional-dependencies]
# GPU installs the default torch build from PyPI (CUDA-enabled by default)
gpu = [
    "nmrglue==0.11",
    "numpy==2.0.2",
    "matplotlib==3.9.3",
    "pandas==2.2.3",
    "tqdm==4.67.1",
    "hydra-core==1.3.2",
    "torch==2.5.1",
    "torchaudio==2.5.1"
]

# CPU needs extra index URL: pip install shimnet[cpu] --extra-index-url https://download.pytorch.org/whl/cpu
cpu = [
    "nmrglue==0.11",
    "numpy==2.0.2",
    "matplotlib==3.9.3",
    "pandas==2.2.3",
    "tqdm==4.67.1",
    "hydra-core==1.3.2",
    "torch==2.5.1+cpu",
    "torchaudio==2.5.1+cpu"
]

# predictions with GUI
gui = [
    "gradio==5.23.2",
    "plotly==6.0.1",
    "ipykernel",
    "psutil",
    "pexpect",
    "requests"
]

# lightweight verion for Colab or another environment with pre-installed torch and common packages
colab = [
    "nmrglue",
    "hydra-core",
]