pyproject.toml

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

[project]
name = "experiment-lab"
description = "A suite of flexible experiments of a variety of different tasks."
keywords = ["experiment", "suite", "lab"]
authors = [{ name = "Rishav Bhagat", email = "rishavbhagat.cs@gmail.com" }]
readme = "README.md"
license = { file = "LICENSE" }
dynamic = ["version"]
requires-python = ">=3.11"
dependencies = [
    "hydra-core",
    "numpy",
    "pandas",
    "torch",
    "wandb",
    "tensorboard",
    "tqdm",
]

[project.urls]
Repository = "https://github.com/rishavb123/ExperimentLab"

[project.optional-dependencies]
all = ["gymnasium", "stable_baselines3", "moviepy", "torchvision", "minigrid"]
rl = ["gymnasium", "stable_baselines3", "minigrid"]
rl-vid = ["moviepy"]
ml = ["torchvision"]
dev = ["pytest", "black", "build", "twine"]

[project.scripts]
run_mc = "experiment_lab.experiments.monte_carlo.main:run_mc_experiment"
run_rl = "experiment_lab.experiments.rl.main:run_rl_experiment"
run_supervised = "experiment_lab.experiments.supervised.main:run_supervised_experiment"

[tool.setuptools.packages.find]
where = ["."]
include = ["experiment_lab*"]
namespaces = false