febolt 0.1.57__tar.gz → 0.1.58__tar.gz

{febolt-0.1.57 → febolt-0.1.58}/Cargo.lock

@@ -362,7 +362,7 @@ checksum = "37909eebbb50d72f9059c3b6d82c0463f2ff062c9e95845c43a6c9c0355411be"
 
 [[package]]
 name = "febolt"
-version = "0.1.57"
+version = "0.1.58"
 dependencies = [
  "blas",
  "intel-mkl-src",

{febolt-0.1.57 → febolt-0.1.58}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "febolt"
-version = "0.1.57"
+version = "0.1.58"
 edition = "2021"
 description = "Statistics library for Python powered by Rust"
 license = "MIT"

{febolt-0.1.57 → febolt-0.1.58}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: febolt
-Version: 0.1.57
+Version: 0.1.58
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{febolt-0.1.57 → febolt-0.1.58}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "febolt"
-version = "0.1.57"
+version = "0.1.58"
 requires-python = ">=3.8"
 description = "A Rust-based Statistics and ML package, callable from Python."
 keywords = ["rust", "python", "Machine Learning", "Statistics", "pyo3"]

{febolt-0.1.57 → febolt-0.1.58}/src/lib.rs

@@ -92,13 +92,23 @@ fn ame<'py>(
     // 1) detect Logit vs Probit
     let is_logit = detect_model_type(model)?;
 
-    // 2) read params
-    let params_pyarray: &PyArray1<f64> = model.getattr("params")?.downcast()?;
-    let beta = unsafe { params_pyarray.as_array() };  // shape(k)
-
-    // 3) read cov
-    let cov_pyarray: &PyArray2<f64> = model.call_method0("cov_params")?.downcast()?;
-    let cov_beta = unsafe { cov_pyarray.as_array() }; // shape(k,k)
+    // 2) read params (handle pandas Series)
+    let params_obj = model.getattr("params")?;
+    let params_pyarray = if let Ok(values) = params_obj.getattr("values") {
+        values.downcast::<PyArray1<f64>>()?
+    } else {
+        params_obj.downcast::<PyArray1<f64>>()?
+    };
+    let beta = unsafe { params_pyarray.as_array() };
+    
+    // 3) read cov (handle pandas DataFrame)
+    let cov_obj = model.call_method0("cov_params")?;
+    let cov_pyarray = if let Ok(values) = cov_obj.getattr("values") {
+        values.downcast::<PyArray2<f64>>()?
+    } else {
+        cov_obj.downcast::<PyArray2<f64>>()?
+    };
+    let cov_beta = unsafe { cov_pyarray.as_array() };
 
     // 4) Get model object and handle exog (X) and exog_names
     let model_obj = model.getattr("model").unwrap_or(model);