[project]
name = "JANUS"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
  "dpkt>=1.9.8",
  "matplotlib>=3.10.8",
  "numpy>=2.4.3",
  "pyyaml>=6.0",
  "scapy>=2.7.0",
  "scikit-learn>=1.8.0",
  "torch>=2.9.1",
  "torchvision>=0.24.1",
  "torchdiffeq>=0.2.5",
  "mamba-ssm>=2.3.1",
  "causal-conv1d>=1.6.1",
  "pandas>=3.0.2",
  "umap-learn>=0.5.12",
  "pyarrow>=24.0.0",
]

[build-system]
requires = ["setuptools>=68"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
where = ["src"]

[tool.uv]
package = false
no-build-isolation-package = ["mamba-ssm", "causal-conv1d"]

[tool.uv.sources]
torch = [
    { index = "pytorch-cu128" },
]
torchvision = [
    { index = "pytorch-cu128" },
]

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true

[dependency-groups]
dev = [
    "pytest>=9.0.3",
]