[package]
edition = "2024"
name = "burn-dataset"
version = "0.20.1"
authors = ["nathanielsimard <nathaniel.simard.42@gmail.com>"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Library with simple dataset APIs for creating ML data pipelines"
documentation = "https://docs.rs/burn-dataset"
readme = "README.md"
keywords = [
"deep-learning",
"machine-learning",
"data",
]
categories = ["science"]
license = "MIT OR Apache-2.0"
repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-dataset"
resolver = "2"
[package.metadata.cargo-udeps.ignore]
normal = [
"strum",
"strum_macros",
]
[package.metadata.docs.rs]
features = ["doc"]
rustdoc-args = [
"--cfg",
"docsrs",
]
[features]
__sqlite-shared = [
"dep:r2d2",
"dep:r2d2_sqlite",
"dep:serde_rusqlite",
"dep:image",
"dep:gix-tempfile",
]
audio = ["hound"]
builtin-sources = [
"vision",
"dep:tar",
"nlp",
]
dataframe = [
"dep:polars",
"dep:planus",
]
default = ["sqlite-bundled"]
doc = ["default"]
fake = ["dep:fake"]
network = ["dep:burn-std"]
nlp = [
"dep:zip",
"dep:encoding_rs",
]
sqlite = [
"__sqlite-shared",
"dep:rusqlite",
]
sqlite-bundled = [
"__sqlite-shared",
"rusqlite/bundled",
]
tracing = ["burn-std/tracing"]
vision = [
"dep:flate2",
"dep:globwalk",
"dep:image",
"network",
]
[lib]
name = "burn_dataset"
path = "src/lib.rs"
[[example]]
name = "hf_dataset"
path = "examples/hf_dataset.rs"
[[example]]
name = "speech_commands"
path = "examples/speech_commands.rs"
[dependencies.burn-std]
version = "0.20.1"
features = ["network"]
optional = true
[dependencies.csv]
version = "1.3.1"
[dependencies.derive-new]
version = "0.7.0"
default-features = false
[dependencies.dirs]
version = "6.0.0"
[dependencies.encoding_rs]
version = "0.8.33"
optional = true
[dependencies.fake]
version = "4.4.0"
optional = true
[dependencies.flate2]
version = "1.1.4"
optional = true
[dependencies.gix-tempfile]
version = "20.0.0"
features = ["signals"]
optional = true
[dependencies.globwalk]
version = "0.9.1"
optional = true
[dependencies.hound]
version = "3.5.1"
optional = true
[dependencies.image]
version = "0.25.9"
optional = true
[dependencies.planus]
version = "=1.1"
optional = true
[dependencies.polars]
version = "0.51.0"
features = ["lazy"]
optional = true
[dependencies.r2d2]
version = "0.8.10"
optional = true
[dependencies.r2d2_sqlite]
version = "0.31.0"
optional = true
[dependencies.rand]
version = "0.9.2"
features = [
"std",
"os_rng",
]
default-features = false
[dependencies.rmp-serde]
version = "1.3.1"
default-features = false
[dependencies.rusqlite]
version = "0.37.0"
optional = true
[dependencies.sanitize-filename]
version = "0.6.0"
[dependencies.serde]
version = "1.0.228"
features = [
"derive",
"alloc",
"std",
"derive",
]
default-features = false
[dependencies.serde_json]
version = "1.0.148"
features = ["std"]
default-features = false
[dependencies.serde_rusqlite]
version = "0.40.0"
optional = true
[dependencies.strum]
version = "0.27.2"
features = ["derive"]
[dependencies.tar]
version = "0.4.44"
optional = true
[dependencies.tempfile]
version = "3.24.0"
[dependencies.thiserror]
version = "2"
default-features = false
[dependencies.zip]
version = "7.0.0"
optional = true
[dev-dependencies.fake]
version = "4.4.0"
[dev-dependencies.rayon]
version = "1.10.0"
[dev-dependencies.rstest]
version = "0.26.1"
[lints.clippy]
[lints.rustdoc]
broken_intra_doc_links = "deny"
invalid_html_tags = "deny"