-
Notifications
You must be signed in to change notification settings - Fork 1
/
Cargo.toml
56 lines (47 loc) · 1.52 KB
/
Cargo.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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
[package]
name = "rl"
version = "0.4.0"
edition = "2021"
description = "A reinforcement learning library"
license = "MIT"
keywords = ["rl", "ml", "machine-learning", "ai"]
categories = ["science"]
repository = "https://github.com/benbaarber/rl"
rust-version = "1.79"
[features]
gym = ["dep:gym-rs", "dep:strum"]
viz = ["dep:ratatui", "dep:crossterm", "dep:tui-logger", "dep:unicode-width"]
[dependencies]
burn = { version = "0.13.2", features = ["autodiff"] }
crossterm = { version = "0.27.0", optional = true }
gym-rs = { version = "0.3.0", git = "https://github.com/MathisWellmann/gym-rs.git", optional = true }
log = { version = "0.4.21", features = ["std"] }
rand = { version = "0.8.5", features = ["alloc"] }
rand_distr = "0.4.3"
ratatui = { version = "0.26.3", features = ["unstable-widget-ref"], optional = true }
strum = { version = "0.26.2", features = ["derive"], optional = true }
tui-logger = { version = "0.11.1", optional = true }
unicode-width = { version = "0.1.13", optional = true }
[dev-dependencies]
burn = { version = "0.13.2", features = ["wgpu", "ndarray"] }
csv = "1.3.0"
gix-fs = "0.11.0"
once_cell = "1.19.0"
statrs = "0.17.1"
strum = "0.26.2"
[[example]]
name = "q_table_frozen_lake"
required-features = ["gym", "viz"]
[[example]]
name = "q_table_snake"
required-features = ["gym", "viz"]
[[example]]
name = "dqn_cartpole"
required-features = ["gym", "viz"]
[[example]]
name = "ten_armed_testbed"
required-features = ["gym"]
[[example]]
name = "policy_iteration_car_rental"
[[example]]
name = "sarsa_windy_gridworld"