57 lines
1.4 KiB
Plaintext
57 lines
1.4 KiB
Plaintext
[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"] |