Cargo.toml

[package]
name = "ai-dataloader"
version = "0.6.2"
authors = ["Tudy Gourmelen <gourmelentudy@gmail.com>"]
edition = "2021"
description = "Rust implementation to the PyTorch DataLoader"
readme = "README.md"
repository = "https://github.com/Tudyx/ai-dataloader"
documentation = "https://docs.rs/ai-dataloader/"
license = "MIT OR Apache-2.0"
keywords = ["dataloader", "pytorch", "tensorflow", "AI", "machine_learning"]
categories = ["science"]
rust-version = "1.63"

[lib]
bench = false

[package.metadata.docs.rs]
all-features = true
# tch doc are broken without this feature.
features = ["tch/doc-only"]
rustdoc-args = ["--cfg", "docsrs"]


[features]
default = ["rayon"]
rayon = ["dep:rayon", "dep:once_cell"]

[dependencies]
ndarray = { version = "0.15.4", features = ["serde"] }
ndarray-rand = "0.14.0"
itertools = "0.13.0"
rand = "0.8.5"
tch = { version = "0.18.0", optional = true, features = ["download-libtorch"] }
rayon = { version = "1.7.0", optional = true }
once_cell = { version = "1.17.1", optional = true }


[dev-dependencies]
criterion = { version = "0.5.1", features = ["html_reports"] }
csv = "1.1.6"
image = "0.25.1"
nshare = { version = "0.9.0", features = ["ndarray", "image"] }

[[example]]
name = "iterable"
path = "examples/iterable.rs"

[[bench]]
name = "throughput"
harness = false

[lints.rust]
rust_2018_idioms = { level = "warn", priority = -1 }
rust_2024_compatibility = { level = "warn", priority = -1 }
missing_docs = "warn"
missing_debug_implementations = "warn"

[lints.clippy]
all = { level = "warn", priority = -1 }
cargo = { level = "warn", priority = -1 }
pedantic = { level = "warn", priority = -1 }
clone_on_ref_ptr = "warn"
fn_params_excessive_bools = "warn"
module_name_repetitions = "allow"