[package]
name = "rtx-sklearn-py"
version = "0.1.0"
edition = "2021"
authors = ["RustyTorch++ Team"]
license = "Apache-2.0"
description = "Python bindings for RustyTorch++ classical ML algorithms with scikit-learn compatibility"
repository = "https://github.com/rustytorch/rustytorch"

[lib]
name = "rustytorch_ml"
crate-type = ["cdylib"]

[dependencies]
# Workspace dependencies
rtx-ml-classic = { path = "../rtx-ml-classic" }
rtx-preprocessing = { path = "../rtx-preprocessing" }
rtx-validation = { path = "../rtx-validation" }
rtx-tensor = { path = "../rtx-tensor" }

# PyO3 for Python bindings
pyo3 = { version = "0.20", features = ["extension-module", "multiple-pymethods", "abi3-py38"] }
pyo3-asyncio = { version = "0.20", features = ["tokio-runtime"], optional = true }

# Numerical computing
numpy = "0.20"
ndarray = "0.15"

# Async support
tokio = { workspace = true }
async-trait = { workspace = true }

# Error handling
anyhow = { workspace = true }
thiserror = { workspace = true }

# Serialization
serde = { workspace = true }
serde_json = { workspace = true }

# Utilities
dashmap = { workspace = true }
parking_lot = { workspace = true }
rand = "0.8"
num_cpus = "1.16"

[build-dependencies]
pyo3-build-config = "0.20"

[dev-dependencies]
proptest = { workspace = true }
tokio-test = { workspace = true }

[features]
default = ["gpu", "async"]
gpu = []
async = ["pyo3-asyncio"]