omar-rq/pyproject.toml at main · MTG/omar-rq · 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
[project]
version = "0.2.1"
name = "omar-rq"
description = "Wrapper library to use the OMAR-RQ models for audio music representation"
readme = "README.md"
license = { text = "AGPL-3.0 license" }
authors = [
  { name = "Pablo Alonso", email = "pablo.alonso@upf.edu" },
  { name = "Pedro Ramoneda", email = "pedro.ramoneda@upf.edu" },
  { name = "R. Oguz Araz", email = "recepoguz.araz@upf.edu" },
]
keywords = [
  "deep learning",
  "audio",
  "transformers",
  "mir",
  "music",
  "representation learning",
  "self-supervised learning",
]
classifiers = [
  "Intended Audience :: Developers",
  "Intended Audience :: Science/Research",

  "License :: OSI Approved :: GNU Affero General Public License v3",

  "Topic :: Scientific/Engineering :: Artificial Intelligence",
  "Topic :: Software Development :: Libraries",
  "Topic :: Multimedia :: Sound/Audio :: Analysis",

  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
]
requires-python = ">= 3.10"
dependencies = [
  "lightning>=2.3,<3.0",
  "torch>=2.6,<3.0",
  "torchaudio>=2.6,<3.0",
  "gin-config>=0.5,<1.0",
  "einops>=0.8,<1.0",
  "transformers>=4.50,<5.0",
]

[project.optional-dependencies]
dev = ["ruff~=0.14", "pytest"]
train = [
  "bayesian_optimization~=1.5.1",
  "matplotlib",
  "scikit_learn~=1.5.2",
  "scipy~=1.14.1",
  "wandb",
]

[project.urls]
Issues = "https://https://github.com/MTG/omar-rq/issues/"
Homepage = "https://github.com/username/omar-rq/"

[build-system]
requires = ["setuptools>=65.5.0", "wheel"]
build-backend = "setuptools.build_meta"