-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
31 lines (25 loc) · 796 Bytes
/
pyproject.toml
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
[tool.poetry]
name = "hcam-torch"
version = "0.0.1"
description = "Implementation of the paper Towards mental time travel: a hierarchical memory for reinforcement learning agents using CleanRL."
authors = ["Euijin Jeong <jin.Prelude@gmail.com>"]
license="MIT"
readme = "README.md"
[tool.poetry.dependencies]
python = "^3.10"
tensorboard = "^2.13.0"
wandb = "^0.15.5"
torch = "^2.0.1"
gymnasium = "^0.27.1"
gym-balletenv = {git = "https://github.com/jinPrelude/gym-balletenv.git"}
# for pytest
pytest = "^7.4.0"
[tool.poetry.group.atari]
optional = true
[tool.poetry.group.atari.dependencies]
envpool = "^0.8.1"
opencv-python = "^4.7.0.68"
gymnasium = {extras = ["accept-rom-license"], version = "^0.27.1"}
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"