predikit 0.4.0__tar.gz → 0.4.1__tar.gz

predikit-0.4.1/.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,38 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: ''
+labels: ''
+assignees: ''
+
+---
+
+**Describe the bug**
+A clear and concise description of what the bug is.
+
+**To Reproduce**
+Steps to reproduce the behavior:
+1. Go to '...'
+2. Click on '....'
+3. Scroll down to '....'
+4. See error
+
+**Expected behavior**
+A clear and concise description of what you expected to happen.
+
+**Screenshots**
+If applicable, add screenshots to help explain your problem.
+
+**Desktop (please complete the following information):**
+ - OS: [e.g. iOS]
+ - Browser [e.g. chrome, safari]
+ - Version [e.g. 22]
+
+**Smartphone (please complete the following information):**
+ - Device: [e.g. iPhone6]
+ - OS: [e.g. iOS8.1]
+ - Browser [e.g. stock browser, safari]
+ - Version [e.g. 22]
+
+**Additional context**
+Add any other context about the problem here.

predikit-0.4.1/.github/ISSUE_TEMPLATE/feature_request.md

@@ -0,0 +1,20 @@
+---
+name: Feature request
+about: Suggest an idea for this project
+title: ''
+labels: ''
+assignees: ''
+
+---
+
+**Is your feature request related to a problem? Please describe.**
+A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
+
+**Describe the solution you'd like**
+A clear and concise description of what you want to happen.
+
+**Describe alternatives you've considered**
+A clear and concise description of any alternative solutions or features you've considered.
+
+**Additional context**
+Add any other context or screenshots about the feature request here.

{predikit-0.4.0 → predikit-0.4.1}/.github/workflows/publish.yml

@@ -4,11 +4,15 @@ on:
   release:
     types: [published]
 
+permissions:
+  contents: read
+
 jobs:
   publish:
     runs-on: ubuntu-latest
     environment: pypi
     permissions:
+      contents: read
       id-token: write
 
     steps:

{predikit-0.4.0 → predikit-0.4.1}/.github/workflows/test.yml

@@ -6,7 +6,31 @@ on:
   pull_request:
     branches: [main]
 
+permissions:
+  contents: read
+
 jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install dev dependencies
+        run: pip install -e ".[dev]"
+
+      - name: Ruff lint
+        run: ruff check src/ tests/
+
+      - name: Ruff format check
+        run: ruff format --check src/ tests/
+
+      - name: Mypy
+        run: mypy src/predikit
+
   test:
     runs-on: ubuntu-latest
     strategy:

predikit-0.4.1/.pre-commit-config.yaml

@@ -0,0 +1,15 @@
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.5.0
+    hooks:
+      - id: ruff
+        args: [--fix]
+      - id: ruff-format
+
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.10.0
+    hooks:
+      - id: mypy
+        additional_dependencies:
+          - pydantic>=2.0
+          - numpy>=1.24

predikit-0.4.1/CHANGELOG.md

@@ -0,0 +1,39 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## [Unreleased]
+
+## [0.4.1] - 2026-06-02
+
+### Added
+- `ruff` (lint + format) and `mypy` (type checking) configured in `pyproject.toml`
+- `.pre-commit-config.yaml` — ruff and mypy hooks run automatically before every commit
+- Lint CI job in GitHub Actions — runs `ruff check`, `ruff format --check`, and `mypy` on every push and PR
+- `CONTRIBUTING.md` — development setup, code style, and PR guidelines
+- `CHANGELOG.md`
+- `CLAUDE.md` — project context for Claude Code
+
+### Changed
+- Bumped version to `0.4.1`
+- Added `ruff>=0.4.0`, `mypy>=1.10`, and `pre-commit>=3.0` to `[dev]` extras
+
+## [0.4.0] - 2026-05-01
+
+### Added
+- `ModelEnsemble` with `weighted_mean` and `weighted_vote` strategies
+- `ainvoke()` async wrapper on `ModelTool` (runs blocking predict in a thread pool)
+- Verbose logging via `verbose=True` on `ModelTool`
+- Snowflake Model Registry loader (`from_snowflake`)
+- MLflow Model Registry loader (`from_mlflow`)
+- `predikit inspect` CLI command
+
+### Changed
+- Moved to `src/` layout with hatchling build backend
+- Upgraded Pydantic dependency to v2
+
+## [0.3.x] and earlier
+
+See [GitHub Releases](https://github.com/Tejas-TA/predikit/releases) for earlier history.

predikit-0.4.1/CLAUDE.md

@@ -0,0 +1,71 @@
+# predikit — Claude Code context
+
+## What this is
+
+predikit wraps fitted scikit-learn / XGBoost models as LLM-callable tools with auto-generated JSON schemas, typed I/O via Pydantic, and first-class support for OpenAI function calling and LangChain.
+
+## Project layout
+
+```
+src/predikit/          Core package (installed as `predikit`)
+  tool.py              ModelTool — main public class
+  registry.py          ToolRegistry — groups multiple tools for bulk export
+  ensemble.py          ModelEnsemble — multi-model reconciliation strategies
+  exceptions.py        LowConfidenceError
+  coerce.py            LLM-string → Python type coercion (bool, int, float, str)
+  introspect.py        Extract sklearn model metadata (feature names, task type)
+  cli.py               `predikit inspect` CLI (click + joblib)
+  exporters/
+    openai.py          .to_openai() — OpenAI function-calling schema
+    langchain.py       .to_langchain() — LangChain StructuredTool
+  loaders/
+    mlflow.py          from_mlflow() — load from MLflow Model Registry
+    snowflake.py       from_snowflake() — load from Snowflake Model Registry
+tests/                 pytest suite (mirrors src/ layout)
+examples/              Standalone usage scripts (01–07)
+```
+
+## Commands
+
+```bash
+# Install for development
+pip install -e ".[dev]"
+
+# Run tests
+pytest --cov=src/predikit --cov-report=term-missing
+
+# Lint (check only)
+ruff check src/ tests/
+
+# Format (check only)
+ruff format --check src/ tests/
+
+# Lint + format (auto-fix)
+ruff check --fix src/ tests/
+ruff format src/ tests/
+
+# Type check
+mypy src/predikit
+
+# All checks via pre-commit
+pre-commit run --all-files
+
+# Install pre-commit hooks
+pre-commit install
+```
+
+## Key conventions
+
+- **Public API** — `ModelTool`, `ToolRegistry`, `ModelEnsemble`, `LowConfidenceError` are all re-exported from `predikit/__init__.py`. Everything else is internal.
+- **Optional imports** — heavy optional deps (`langchain`, `xgboost`, `mlflow`, `snowflake`) are imported inside functions/methods, never at module level, so missing extras don't break core imports.
+- **Feature name matching** — inputs map to model features by column name (using `feature_names_in_` from sklearn), not by position. Mismatches raise a `ValueError` with a clear diff.
+- **Bool coercion** — `coerce_value()` converts LLM strings (`"yes"`, `"true"`, `"1"`, `"on"`) to Python bools before Pydantic validation.
+- **Async** — `ainvoke()` is a thin wrapper over `invoke()` via `loop.run_in_executor`. No async-native I/O involved.
+- **Ensemble strategies** — `"collect"`, `"mean"`, `"vote"`, `"weighted_mean"`, `"weighted_vote"`. Weighted strategies take a `weights` list parallel to `tools`.
+
+## Release process
+
+1. Bump `version` in `pyproject.toml` and `__version__` in `src/predikit/__init__.py`.
+2. Add a section to `CHANGELOG.md`.
+3. Commit, push to `main`, create a GitHub Release with the version tag (e.g. `v0.4.1`).
+4. `publish.yml` auto-triggers a PyPI upload via OIDC trusted publishing.

predikit-0.4.1/CONTRIBUTING.md

@@ -0,0 +1,66 @@
+# Contributing to predikit
+
+Thanks for your interest in contributing.
+
+## Prerequisites
+
+- Python 3.10+
+- [pre-commit](https://pre-commit.com/) (`pip install pre-commit`)
+
+## Development setup
+
+```bash
+git clone https://github.com/Tejas-TA/predikit
+cd predikit
+pip install -e ".[dev]"
+pre-commit install
+```
+
+`pre-commit install` wires up ruff and mypy to run automatically before every commit.
+
+## Running tests
+
+```bash
+pytest --cov=src/predikit --cov-report=term-missing
+```
+
+## Code style
+
+This project uses [ruff](https://docs.astral.sh/ruff/) for linting and formatting, and [mypy](https://mypy.readthedocs.io/) for type checking. All three are configured in `pyproject.toml`.
+
+```bash
+# Check and auto-fix
+ruff check --fix src/ tests/
+ruff format src/ tests/
+
+# Type check
+mypy src/predikit
+```
+
+Or run everything at once via pre-commit:
+
+```bash
+pre-commit run --all-files
+```
+
+CI enforces all three checks on every push and PR (see `.github/workflows/test.yml`).
+
+## Pull request guidelines
+
+- Target `main`.
+- Keep changes focused — one feature or fix per PR.
+- Add or update tests for any changed behaviour.
+- Run `pre-commit run --all-files` before pushing; CI will catch failures regardless.
+
+## Adding a new exporter or loader
+
+- Exporters live in `src/predikit/exporters/` and should expose a single function consumed by `ModelTool`.
+- Loaders live in `src/predikit/loaders/` and should return a `ModelTool` instance.
+- Any new optional dependency must be added to both the relevant optional group and `dev` in `pyproject.toml`, and imported inside the function body (not at module level).
+
+## Release process (maintainers)
+
+1. Bump `version` in `pyproject.toml` and `__version__` in `src/predikit/__init__.py`.
+2. Add a section to `CHANGELOG.md`.
+3. Merge to `main`, create a GitHub Release tagged `vX.Y.Z`.
+4. `publish.yml` automatically publishes to PyPI via OIDC trusted publishing.

{predikit-0.4.0 → predikit-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: predikit
-Version: 0.4.0
+Version: 0.4.1
 Summary: Turn any trained sklearn/XGBoost model into an LLM-callable tool with auto-generated schemas and typed I/O.
 Project-URL: Homepage, https://github.com/Tejas-TA/predikit
 Project-URL: Repository, https://github.com/Tejas-TA/predikit
@@ -31,9 +31,12 @@ Requires-Dist: click>=8.0; extra == 'dev'
 Requires-Dist: joblib>=1.2; extra == 'dev'
 Requires-Dist: langchain-core>=0.1; extra == 'dev'
 Requires-Dist: mlflow>=2.0; extra == 'dev'
+Requires-Dist: mypy>=1.10; extra == 'dev'
 Requires-Dist: pandas>=1.5; extra == 'dev'
+Requires-Dist: pre-commit>=3.0; extra == 'dev'
 Requires-Dist: pytest-cov>=4.0; extra == 'dev'
 Requires-Dist: pytest>=7.0; extra == 'dev'
+Requires-Dist: ruff>=0.4.0; extra == 'dev'
 Requires-Dist: xgboost>=1.7; extra == 'dev'
 Provides-Extra: langchain
 Requires-Dist: langchain-core>=0.1; extra == 'langchain'
@@ -49,9 +52,23 @@ Description-Content-Type: text/markdown
 [![PyPI version](https://img.shields.io/pypi/v/predikit.svg)](https://pypi.org/project/predikit/)
 [![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-green.svg)](LICENSE)
+[![CI](https://github.com/Tejas-TA/predikit/actions/workflows/test.yml/badge.svg)](https://github.com/Tejas-TA/predikit/actions/workflows/test.yml)
+[![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
 
+### 📈 Project Traffic
+Detailed breakdown of downloads by version, region, and platform:
 
-Turn any trained scikit-learn or XGBoost model into an LLM-callable tool — auto-generated JSON schemas, typed I/O, zero boilerplate.
+[![Downloads](https://pepy.tech/badge/predikit?style=for-the-badge)](https://pepy.tech/project/predikit)
+
+## Table of Contents
+- [Install](#install)
+- [30-second example](#30-second-example)
+- [Core API](#core-api)
+- [Cookbook](#cookbook)
+- [Contributing](#contributing)
+- [License](#license)
+
+## Turn any trained scikit-learn or XGBoost model into an LLM-callable tool — auto-generated JSON schemas, typed I/O, zero boilerplate.
 
 ```python
 tool = ModelTool(model=clf, name="classify_iris", ...)
@@ -333,7 +350,7 @@ tool = from_snowflake(
     model_name="VACATION_CHURN",
     model_version="V3",
     name="churn_risk",
-    description="Vacation ownership churn classifier.",
+    description="Churn classifier.",
     input_schema=MemberInput,
     output_name="churn_probability",
     output_description="Churn probability 0–1",
@@ -351,10 +368,13 @@ See [`examples/03_orlando_real_estate.py`](examples/03_orlando_real_estate.py) f
 
 Planned for later releases:
 
-- HuggingFace / PyTorch / TensorFlow support
-- Async invocation
-- Weighted ensemble strategies
-- CLI (`predikit inspect model.pkl`)
+- HuggingFace / PyTorch / TensorFlow model support
+- Streaming inference support
+- OpenAI Assistants API integration
+
+## Contributing
+
+See [CONTRIBUTING.md](CONTRIBUTING.md) for development setup, code style, and PR guidelines. The [CHANGELOG](CHANGELOG.md) tracks notable changes per release.
 
 ## License
 

{predikit-0.4.0 → predikit-0.4.1}/README.md

@@ -2,9 +2,23 @@
 [![PyPI version](https://img.shields.io/pypi/v/predikit.svg)](https://pypi.org/project/predikit/)
 [![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/)
 [![License: MIT](https://img.shields.io/badge/License-MIT-green.svg)](LICENSE)
+[![CI](https://github.com/Tejas-TA/predikit/actions/workflows/test.yml/badge.svg)](https://github.com/Tejas-TA/predikit/actions/workflows/test.yml)
+[![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff)
 
+### 📈 Project Traffic
+Detailed breakdown of downloads by version, region, and platform:
 
-Turn any trained scikit-learn or XGBoost model into an LLM-callable tool — auto-generated JSON schemas, typed I/O, zero boilerplate.
+[![Downloads](https://pepy.tech/badge/predikit?style=for-the-badge)](https://pepy.tech/project/predikit)
+
+## Table of Contents
+- [Install](#install)
+- [30-second example](#30-second-example)
+- [Core API](#core-api)
+- [Cookbook](#cookbook)
+- [Contributing](#contributing)
+- [License](#license)
+
+## Turn any trained scikit-learn or XGBoost model into an LLM-callable tool — auto-generated JSON schemas, typed I/O, zero boilerplate.
 
 ```python
 tool = ModelTool(model=clf, name="classify_iris", ...)
@@ -286,7 +300,7 @@ tool = from_snowflake(
     model_name="VACATION_CHURN",
     model_version="V3",
     name="churn_risk",
-    description="Vacation ownership churn classifier.",
+    description="Churn classifier.",
     input_schema=MemberInput,
     output_name="churn_probability",
     output_description="Churn probability 0–1",
@@ -304,10 +318,13 @@ See [`examples/03_orlando_real_estate.py`](examples/03_orlando_real_estate.py) f
 
 Planned for later releases:
 
-- HuggingFace / PyTorch / TensorFlow support
-- Async invocation
-- Weighted ensemble strategies
-- CLI (`predikit inspect model.pkl`)
+- HuggingFace / PyTorch / TensorFlow model support
+- Streaming inference support
+- OpenAI Assistants API integration
+
+## Contributing
+
+See [CONTRIBUTING.md](CONTRIBUTING.md) for development setup, code style, and PR guidelines. The [CHANGELOG](CHANGELOG.md) tracks notable changes per release.
 
 ## License
 

{predikit-0.4.0 → predikit-0.4.1}/examples/03_orlando_real_estate.py

@@ -5,7 +5,7 @@ Trains an XGBoost model on synthetic Orlando-area housing data,
 wraps it as an LLM-callable tool, and shows the full end-to-end flow:
 schema generation → registry export → direct invocation → callable.
 
-Requires: pip install modelbridge[xgboost]
+Requires: pip install xgboost
 """
 import json
 
@@ -17,7 +17,7 @@ from sklearn.model_selection import train_test_split
 try:
     from xgboost import XGBRegressor
 except ImportError:
-    raise SystemExit("XGBoost not installed. Run: pip install modelbridge[xgboost]")
+    raise SystemExit("XGBoost not installed. Run: pip install xgboost")
 
 from predikit import ModelTool, ToolRegistry
 

{predikit-0.4.0 → predikit-0.4.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "predikit"
-version = "0.4.0"
+version = "0.4.1"
 description = "Turn any trained sklearn/XGBoost model into an LLM-callable tool with auto-generated schemas and typed I/O."
 readme = "README.md"
 license = {text = "MIT"}
@@ -45,6 +45,9 @@ dev = [
     "mlflow>=2.0",
     "click>=8.0",
     "joblib>=1.2",
+    "ruff>=0.4.0",
+    "mypy>=1.10",
+    "pre-commit>=3.0",
 ]
 
 [project.scripts]
@@ -61,3 +64,20 @@ packages = ["src/predikit"]
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]
+
+[tool.ruff]
+line-length = 100
+target-version = "py310"
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "UP", "B", "SIM"]
+ignore = ["E501"]
+
+[tool.ruff.lint.isort]
+known-first-party = ["predikit"]
+
+[tool.mypy]
+python_version = "3.10"
+warn_return_any = true
+warn_unused_configs = true
+ignore_missing_imports = true

{predikit-0.4.0 → predikit-0.4.1}/src/predikit/__init__.py

@@ -1,7 +1,7 @@
-from .tool import ModelTool
-from .registry import ToolRegistry
 from .ensemble import ModelEnsemble
 from .exceptions import LowConfidenceError
+from .registry import ToolRegistry
+from .tool import ModelTool
 
 __all__ = ["ModelTool", "ToolRegistry", "ModelEnsemble", "LowConfidenceError"]
-__version__ = "0.4.0"
+__version__ = "0.4.1"

{predikit-0.4.0 → predikit-0.4.1}/src/predikit/cli.py

@@ -1,31 +1,41 @@
 from __future__ import annotations
+
 import json
 import sys
 
 try:
     import click
+
     _CLICK_AVAILABLE = True
 except ImportError:
     _CLICK_AVAILABLE = False
 
 
 if _CLICK_AVAILABLE:
+
     @click.group()
     def cli() -> None:
         """predikit — ML model utilities for LLM agents."""
 
     @cli.command()
     @click.argument("model_path", type=click.Path(exists=True))
-    @click.option("--name", default="model", show_default=True, help="Tool name used in schema generation.")
-    @click.option("--description", default="ML model prediction", show_default=True, help="Tool description.")
+    @click.option(
+        "--name", default="model", show_default=True, help="Tool name used in schema generation."
+    )
+    @click.option(
+        "--description", default="ML model prediction", show_default=True, help="Tool description."
+    )
     def inspect(model_path: str, name: str, description: str) -> None:
         """Inspect a saved model file and print its metadata and OpenAI schema."""
         try:
             import joblib
-        except ImportError:
-            raise click.ClickException("joblib is required. Install it with: pip install predikit[cli]")
+        except ImportError as err:
+            raise click.ClickException(
+                "joblib is required. Install it with: pip install predikit[cli]"
+            ) from err
 
         from pydantic import create_model
+
         from .introspect import introspect
         from .tool import ModelTool
 
@@ -64,9 +74,13 @@ if _CLICK_AVAILABLE:
             click.echo("\nOpenAI schema: unavailable (fit model with a named DataFrame to enable)")
 
 else:
+
     def cli() -> None:  # type: ignore[misc]
         """Fallback when click is not installed."""
-        print("Error: 'click' is required. Install it with: pip install predikit[cli]", file=sys.stderr)
+        print(
+            "Error: 'click' is required. Install it with: pip install predikit[cli]",
+            file=sys.stderr,
+        )
         sys.exit(1)
 
 

{predikit-0.4.0 → predikit-0.4.1}/src/predikit/coerce.py

@@ -1,6 +1,6 @@
 from typing import Any
-from pydantic import BaseModel
 
+from pydantic import BaseModel
 
 _BOOL_TRUE = {"true", "1", "yes", "on"}
 _BOOL_FALSE = {"false", "0", "no", "off"}
@@ -18,7 +18,9 @@ def coerce_value(value: Any, target_type: type) -> Any:
                 return True
             if low in _BOOL_FALSE:
                 return False
-            raise ValueError(f"Cannot interpret {value!r} as bool. Expected one of: true/false, yes/no, 1/0, on/off")
+            raise ValueError(
+                f"Cannot interpret {value!r} as bool. Expected one of: true/false, yes/no, 1/0, on/off"
+            )
         return bool(value)
 
     if target_type is int:

{predikit-0.4.0 → predikit-0.4.1}/src/predikit/ensemble.py

@@ -1,11 +1,13 @@
 from __future__ import annotations
+
 import asyncio
 from collections import Counter
-from typing import Any, Callable
+from collections.abc import Callable
+from typing import Any
 
-from .tool import ModelTool
-from .exporters.openai import to_openai_schema
 from .exporters.langchain import to_langchain_tool
+from .exporters.openai import to_openai_schema
+from .tool import ModelTool
 
 _VALID_STRATEGIES = {"collect", "mean", "weighted_mean", "vote", "weighted_vote"}
 
@@ -71,14 +73,14 @@ class ModelEnsemble:
         if self.strategy == "weighted_mean":
             numeric = [float(v) for v in values]
             total = sum(weights)
-            return {output_name: sum(w * v for w, v in zip(weights, numeric)) / total}
+            return {output_name: sum(w * v for w, v in zip(weights, numeric, strict=True)) / total}
 
         if self.strategy == "vote":
             return {output_name: Counter(values).most_common(1)[0][0]}
 
         if self.strategy == "weighted_vote":
             tally: dict[Any, float] = {}
-            for w, v in zip(weights, values):
+            for w, v in zip(weights, values, strict=True):
                 tally[v] = tally.get(v, 0.0) + w
             return {output_name: max(tally, key=tally.__getitem__)}
 
@@ -88,12 +90,13 @@ class ModelEnsemble:
     def to_openai(self) -> dict:
         return to_openai_schema(self)
 
-    def to_langchain(self):
+    def to_langchain(self) -> Any:
         return to_langchain_tool(self)
 
     def to_callable(self) -> Callable[..., dict]:
         def _fn(**kwargs) -> dict:
             return self.invoke(kwargs)
+
         _fn.__name__ = self.name
         _fn.__doc__ = self.description
         return _fn

{predikit-0.4.0 → predikit-0.4.1}/src/predikit/exporters/langchain.py

@@ -1,18 +1,19 @@
 from __future__ import annotations
-from typing import TYPE_CHECKING
+
+from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
     from ..tool import ModelTool
 
 
-def to_langchain_tool(tool: ModelTool):
+def to_langchain_tool(tool: ModelTool) -> Any:
     """Convert a ModelTool to a LangChain StructuredTool."""
     try:
         from langchain_core.tools import StructuredTool
-    except ImportError:
+    except ImportError as err:
         raise ImportError(
             "langchain-core is required. Install with: pip install predikit[langchain]"
-        )
+        ) from err
 
     def _run(**kwargs) -> dict:
         return tool.invoke(kwargs)

{predikit-0.4.0 → predikit-0.4.1}/src/predikit/exporters/openai.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:

{predikit-0.4.0 → predikit-0.4.1}/src/predikit/introspect.py

@@ -8,9 +8,7 @@ def introspect(model: Any) -> dict:
     meta["feature_names"] = (
         list(model.feature_names_in_) if hasattr(model, "feature_names_in_") else None
     )
-    meta["n_features"] = (
-        int(model.n_features_in_) if hasattr(model, "n_features_in_") else None
-    )
+    meta["n_features"] = int(model.n_features_in_) if hasattr(model, "n_features_in_") else None
 
     if hasattr(model, "classes_"):
         meta["task"] = "classification"