rfi_toolbox/pyproject.toml at main · preshanth/rfi_toolbox · 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
[build-system]
requires = ["setuptools>=42", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "rfi_toolbox"
version = "0.2.0"
description = "Universal foundation for ML-based RFI detection in radio astronomy"
readme = "README.md"
license = "GPL-3.0-or-later"
authors = [
    { name = "Preshanth Jagannathan", email = "pjaganna@nrao.edu" }
]
dependencies = [
    "numpy",
    "scipy",
    "tqdm",
    "pyyaml",
]

[project.optional-dependencies]
# Visualization tools
viz = [
    "bokeh >= 2.4.0",
    "matplotlib >= 3.0.0",
]

# Training and ML operations
training = [
    "torch >= 1.9.0",
    "scikit-learn >= 0.24.0",
]

# CASA tools for Measurement Set I/O
casa = [
    "casatools>=6.5.0",
    "casatasks>=6.5.0",
]

# Development tools
dev = [
    "pytest >= 7.0.0",
    "pytest-cov >= 4.0.0",
    "black >= 23.0.0",
    "ruff >= 0.1.0",
]

# Everything (for convenience)
all = [
    "rfi_toolbox[viz,training,casa,dev]",
]

[project.scripts]
generate_rfi_dataset = "rfi_toolbox.scripts.generate_dataset:main"
train_rfi_model = "rfi_toolbox.scripts.train_model:main"
evaluate_rfi_model = "rfi_toolbox.scripts.evaluate_model:main"
visualize_rfi_data = "rfi_toolbox.visualization.visualize:main"
normalize_rfi_data = "rfi_toolbox.scripts.normalize_rfi_data:main"

[tool.setuptools]
packages = {find = {}}

[tool.setuptools.package-data]
rfi_toolbox = ["py.typed"]