graph_rl/Cargo.toml at master · dashdeckers/graph_rl · GitHub

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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
[package]
name = "graph_rl"
version = "0.1.0"
edition = "2021"
authors = ["Travis Hammond <dashdeckers@gmail.com>"]
repository = "https://github.com/dashdeckers/rl_toolkit"
description = "A toolkit for reinforcement learning research."
keywords = [
    "reinforcement learning",
    "machine-learning",
    "actor-critic",
    "planning",
    "graph",
]
license = "mit"
categories = ["research"]


[dependencies]

# Developer candy
anyhow = "1.0.70"
auto_ops = "0.3.0"
ordered-float = { version = "4.1.1", features = ["serde"] }
unzip-n = "0.1.2"
strum = { version = "0.26.1", features = ["derive"] }

# GUI
eframe = "0.23.0"
egui = "0.23.0"
egui_plot = "0.23.0"

# CLI
clap = { version = "4.2.4", features = ["derive"] }

# Logging
tracing = "0.1.37"
tracing-subscriber = "0.3.17"

# Random
rand_distr = "0.4.3"
rand = { version = "0.8.5", features = ["std"] }

# Graph
petgraph = "0.6.3"

# Dataframes
polars = { version = "0.34.2", features = ["parquet"] }

# Python Interop
pyo3 = { version = "0.20.0", features = ["auto-initialize"] }

# Serialization
serde = { version = "1.0.192", features = ["derive"] }
ron = "0.8.1"

# Docs
simple-mermaid = "0.1.0"


# Enabling katex on docs.rs. To build locally:
# RUSTDOCFLAGS="--html-in-header docs/katex-header.html" cargo doc --no-deps --open
[package.metadata.docs.rs]
rustdoc-args = [
    "--html-in-header",
    "docs/katex-header.html",
]

# Deep Learning (config gate: I didn't install CUDA on my windows machine)
[target.'cfg(windows)'.dependencies]
candle-core = { git = "https://github.com/dashdeckers/candle.git" }
candle-nn = { git = "https://github.com/dashdeckers/candle.git" }

[target.'cfg(not(windows))'.dependencies]
candle-core = { git = "https://github.com/dashdeckers/candle.git", features = ["cuda"] }
candle-nn = { git = "https://github.com/dashdeckers/candle.git", features = ["cuda"] }