bossanova 0.1.0.dev12__tar.gz → 0.1.0.dev13__tar.gz

{bossanova-0.1.0.dev12 → bossanova-0.1.0.dev13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bossanova
-Version: 0.1.0.dev12
+Version: 0.1.0.dev13
 Summary: Bridging statistical cultures with some jazz
 Author: Eshin Jolly
 License-Expression: MIT

{bossanova-0.1.0.dev12 → bossanova-0.1.0.dev13}/bossanova/internal/operations/compare/compare.py

@@ -56,6 +56,7 @@ def compare(
     cv: int | str = 5,
     seed: int | None = None,
     metric: str = "mse",
+    precision: int | None = 4,
 ) -> pl.DataFrame:
     """Compare nested statistical models.
 
@@ -89,6 +90,8 @@ def compare(
         cv: For CV comparison only. Number of folds or "loo" for leave-one-out.
         seed: For CV comparison only. Random seed for reproducible splits.
         metric: For CV comparison only. Error metric ("mse", "rmse", "mae").
+        precision: Number of decimal places to round float columns to.
+            Default is 4. Pass ``None`` to return full (unrounded) precision.
 
     Returns:
         DataFrame with comparison results. Columns depend on method:
@@ -220,14 +223,21 @@ def compare(
 
     # Dispatch to appropriate comparison function
     if method == "f":
-        return compare_f_test(model_list)
+        result = compare_f_test(model_list)
     elif method == "lrt":
-        return compare_lrt(model_list)
+        result = compare_lrt(model_list)
     elif method == "deviance":
-        return compare_deviance(model_list, test=test)
+        result = compare_deviance(model_list, test=test)
     elif method == "cv":
-        return compare_cv(model_list, cv=cv, seed=seed, metric=metric)
+        result = compare_cv(model_list, cv=cv, seed=seed, metric=metric)
     else:
         raise ValueError(
             f"Unknown method: {method}. Use 'auto', 'f', 'lrt', 'deviance', or 'cv'."
         )
+
+    # Round float columns to requested precision
+    if precision is not None:
+        float_cols = [c for c in result.columns if result[c].dtype == pl.Float64]
+        result = result.with_columns(pl.col(float_cols).round(precision))
+
+    return result

{bossanova-0.1.0.dev12 → bossanova-0.1.0.dev13}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "bossanova"
 requires-python = ">=3.11"
-version = "0.1.0.dev12"
+version = "0.1.0.dev13"
 authors = [
     {name = "Eshin Jolly"},
 ]