-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpyproject.toml
More file actions
71 lines (63 loc) · 1.52 KB
/
pyproject.toml
File metadata and controls
71 lines (63 loc) · 1.52 KB
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
[project]
name = "signal-mae"
version = "2.0.0"
description = "Modular audio/signal machine learning framework with plugin-based architecture for training masked autoencoders"
readme = "README.md"
requires-python = ">=3.10"
license = {text = "MIT"}
authors = [
{name = "Louis Aberdeen", email = "[email protected]"}
]
keywords = ["audio", "rf", "signals", "machine-learning", "masked-autoencoder", "pytorch", "fiftyone", "torchsig", "modulation-classification"]
dependencies = [
"torch>=2.0.0",
"torchvision>=0.15.0",
"numpy>=1.24.0",
"pandas>=2.0.0",
"einops>=0.7.0",
"tqdm>=4.65.0",
"pillow>=10.0.0",
]
[project.optional-dependencies]
audio = [
"librosa>=0.10.0",
"scipy>=1.10.0",
]
rf = [
"torchsig>=0.5.0",
"scipy>=1.10.0",
]
visualization = [
"fiftyone>=0.23.0",
"umap-learn>=0.5.0",
"matplotlib>=3.7.0",
]
training = [
"mlflow>=2.0.0",
]
dev = [
"pytest>=7.0.0",
"pytest-cov>=4.0.0",
]
all = [
"signal-mae[audio,rf,visualization,training,dev]",
]
[project.urls]
Homepage = "https://github.com/louisaberdeen/signal-MAE"
Repository = "https://github.com/louisaberdeen/signal-MAE"
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
[tool.hatch.build.targets.wheel]
packages = ["src"]
[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py"]
python_functions = ["test_*"]
addopts = "-v --tb=short"
[tool.ruff]
line-length = 100
target-version = "py310"
[tool.ruff.lint]
select = ["E", "F", "I", "W"]
ignore = ["E501"]