KAMoE/pyproject.toml at main · remigenet/KAMoE · 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
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.poetry]
name = "kamoe"
version = "0.1.8"
description = "Kolmogorov-Arnold Mixtures of Experts"
authors = ["Rémi Genet", "Hugo Inzirillo"]
readme = "README.md"
packages = [{include = "kamoe"}]

[tool.poetry.dependencies]
python = ">=3.9,<3.12"
keras = ">=3.0.0,<4.0"
keras_efficient_kan = "^0.1.4"

[tool.poetry.group.dev.dependencies]
pytest = "^7.4.0"
pytest-xdist = "^3.3.0"
tensorflow = "^2.15.0"
torch = "^2.0.0"
jax = "^0.4.13"
jaxlib = "^0.4.13"

[tool.pytest.ini_options]
addopts = "-v"
testpaths = ["tests"]
filterwarnings = [
    "ignore:Can't initialize NVML:UserWarning",
    "ignore:jax.xla_computation is deprecated:DeprecationWarning",
    "ignore::DeprecationWarning:jax._src.dtypes"
]