import os
from setuptools import setup
from setuptools_rust import Binding, RustExtension
CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
rust_extensions = [
RustExtension(
"outlines_core.outlines_core_rs",
f"{CURRENT_DIR}/Cargo.toml",
binding=Binding.PyO3,
features=["python-bindings"],
rustc_flags=["--crate-type=cdylib"],
),
]
setup(
rust_extensions=rust_extensions,
)