Release v0.1.6

Author: xRiskLab

fastwoe/fastwoe.py

@@ -697,7 +697,7 @@ def fit(
         self.is_multiclass_target = None
 
         self.is_continuous_target = bool(is_continuous)
-        self.is_binary_target = is_binary
+        self.is_binary_target = bool(is_binary)
         self.is_multiclass_target = bool(is_multiclass)
 
         # Update encoder_kwargs based on target type
@@ -1730,7 +1730,7 @@ def _bin_with_tree(
             # Bin the non-missing values
             col_data = X_fit[col]
             if hasattr(col_data, "values"):
-                col_values = col_data.values
+                col_values = np.asarray(col_data.values, dtype=float)
             else:
                 col_values = np.array(col_data)
             # Custom binning logic to handle right-inclusive intervals correctly

fastwoe/logging_config.py

@@ -5,6 +5,8 @@
 logging across the entire codebase.
 """
 
+from typing import Any
+
 _HAS_LOGGING = False
 
 try:
@@ -22,7 +24,7 @@ def _noop(self, *args, **kwargs):
 
         info = debug = warning = error = critical = _noop
 
-    logger = _NullLogger()
+    logger: Any = _NullLogger()  # type: ignore[assignment, no-redef]
 
 
 def setup_logger(level: str = "INFO") -> None:

fastwoe/plots.py

@@ -16,14 +16,17 @@
 if TYPE_CHECKING:
     from matplotlib import pyplot as plt
     from matplotlib.axes import Axes
+    from matplotlib.figure import Figure
 else:
     # Runtime import - needed for actual plotting
     try:
         from matplotlib import pyplot as plt  # noqa: F401
         from matplotlib.axes import Axes  # noqa: F401
+        from matplotlib.figure import Figure  # noqa: F401
     except ImportError:
         plt = None  # type: ignore[assignment]
         Axes = Any  # type: ignore[assignment, misc]
+        Figure = Any  # type: ignore[assignment, misc]
 
 from .metrics import somersd_yx
 
@@ -134,7 +137,11 @@ def plot_performance(
         fig, ax1 = plt.subplots(1, 1, figsize=figsize, dpi=dpi)
     else:
         ax1 = ax
-        fig = ax1.get_figure()
+        fig_raw = ax1.get_figure()
+        if fig_raw is None:
+            raise ValueError("Could not get figure from axes")
+        # Type narrowing for matplotlib figure
+        fig = fig_raw  # type: ignore[assignment, no-redef]
 
     # CAP curve for binary and continuous targets
     n = len(y_true)

pyproject.toml

@@ -172,6 +172,7 @@ dev = [
     "seaborn>=0.11.0",
     "rich>=13.7.0", # Required for pretty printing
     "ty>=0.0.1a21",
+    "mypy>=1.11.0", # Type checker
     "faiss-cpu>=1.12.0", # Required for FAISS tests
     "pandas-stubs>=2.0.0", # Type stubs for pandas (for mypy/type checking)
 ]