pyproject.toml

[tool.poetry]
name = "js-rl"
version = "0.1.0"
description = ""
authors = ["Pranav Maganti <pranav.maganti@gmail.com>"]
readme = "README.md"
packages = [{include = "js_rl"}]

[tool.poetry.dependencies]
python = "^3.11"
esprima = "^4.0.1"
numpy = "^1.24.2"
tqdm = "^4.64.1"
transformers = "^4.27.4"
torch = "^2.0.0"
strenum = "^0.4.10"
gymnasium = "^0.28.1"
esutils = "^1.0.1"
escodegen = "^1.0.11"
seaborn = "^0.12.2"
optimum = "^1.8.7"


[tool.poetry.group.dev.dependencies]
black = {version = "^23.1.0", allow-prereleases = true}
pytest = "^7.2.1"
ipykernel = "^6.22.0"
pandas = "^2.0.1"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.pytest.ini_options]
pythonpath = [
  "src"
]
minversion = "6.0"
addopts = "-ra -q"
testpaths = [
    "tests",
    "integration",
]

[tool.isort]
skip = ["engines"]

[tool.black]
exclude = '''engines'''