SR-MIDAS/pyproject.toml at main · AdvancedPhotonSource/SR-MIDAS · GitHub

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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "sr-midas"
dynamic = ["version"]
description = "Super-resolution CNN workflow for MIDAS high-energy X-ray diffraction data"
readme = "README.md"
license = { file = "LICENSE" }
requires-python = "==3.12.4"
keywords = ["XRD", "HEDM", "super-resolution", "diffraction", "CNN", "MIDAS"]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "Topic :: Scientific/Engineering :: Physics",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
]
dependencies = [
    "numpy==1.26.4",
    "pandas==3.0.2",
    "h5py==3.16.0",
    "zarr==2.18.7",
    "scipy==1.17.1",
    "scikit-image==0.25.2",
    "scikit-learn==1.8.0",
    "matplotlib==3.10.8",
    "tqdm==4.67.3",
    "torch==2.9.1",
    "parsl==2024.9.16",
    "pillow==12.2.0",
    "numba==0.59.1",
    "numcodecs==0.15.1",
    "hdf5plugin==5.0.0",
    "dash==2.18.1",
    "plotly==5.22.0",
    "dash_bootstrap_components==1.6.0",
    "diplib==3.5.1",
    "fsspec==2026.3.0",
    "opencv-python",
    "globus_compute_sdk==2.27.1",
    "globus_sdk==3.63.0",
    "diskcache",
    "multiprocess",
    "pyqtgraph",
    "PyQt5",
    "PyQt6==6.9.0",
]

[project.optional-dependencies]
optuna = ["optuna>=3.0"]
all    = ["sr-midas[optuna]"]
dev    = ["pytest>=7.4", "pytest-cov>=4.1", "ruff>=0.3"]

[project.scripts]
sr-midas-create-peakbank   = "sr_midas._cli.create_peakbank_cli:main"
sr-midas-create-patchstore = "sr_midas._cli.create_patchstore_cli:main"
sr-midas-train             = "sr_midas._cli.train_cli:main"
sr-midas-hp-optimize       = "sr_midas._cli.hp_optimize_cli:main"
sr-midas-predict           = "sr_midas._cli.predict_cli:main"
sr-midas-create-pred-pst   = "sr_midas._cli.create_pred_patchstore_cli:main"
sr-midas-process           = "sr_midas._cli.sr_process_cli:main"

[tool.pytest.ini_options]
testpaths = ["tests"]
markers = [
    "integration: end-to-end test requiring MIDAS data on disk",
]

[tool.hatch.version]
path = "src/sr_midas/_version.py"

[tool.hatch.build.targets.wheel]
packages = ["src/sr_midas"]
# Include non-Python data files (model weights and train args)
artifacts = [
    "src/sr_midas/models/cnnsr/*.json",
    "src/sr_midas/models/cnnsr/pretrained/**/*.pth",
    "src/sr_midas/models/cnnsr/pretrained/**/*.json",
]