predikit 0.3.1__tar.gz → 0.3.2__tar.gz

{predikit-0.3.1 → predikit-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: predikit
-Version: 0.3.1
+Version: 0.3.2
 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
@@ -42,7 +42,6 @@ Description-Content-Type: text/markdown
 
 # predikit
 [![PyPI version](https://img.shields.io/pypi/v/predikit.svg)](https://pypi.org/project/predikit/)
-[![Tests](https://github.com/Tejas-TA/predikit/actions/workflows/test.yml/badge.svg)](https://github.com/Tejas-TA/predikit/actions)
 [![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)
 

{predikit-0.3.1 → predikit-0.3.2}/README.md

@@ -1,6 +1,5 @@
 # predikit
 [![PyPI version](https://img.shields.io/pypi/v/predikit.svg)](https://pypi.org/project/predikit/)
-[![Tests](https://github.com/Tejas-TA/predikit/actions/workflows/test.yml/badge.svg)](https://github.com/Tejas-TA/predikit/actions)
 [![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)
 

{predikit-0.3.1 → predikit-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "predikit"
-version = "0.3.1"
+version = "0.3.2"
 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"}

{predikit-0.3.1 → predikit-0.3.2}/src/predikit/__init__.py

@@ -4,4 +4,4 @@ from .ensemble import ModelEnsemble
 from .exceptions import LowConfidenceError
 
 __all__ = ["ModelTool", "ToolRegistry", "ModelEnsemble", "LowConfidenceError"]
-__version__ = "0.3.1"
+__version__ = "0.3.2"

{predikit-0.3.1 → predikit-0.3.2}/tests/test_loaders_mlflow.py

@@ -1,7 +1,6 @@
 """Integration tests for from_mlflow: logs a real model, loads, invokes."""
 import os
 import tempfile
-
 import numpy as np
 import pytest
 from pydantic import BaseModel, Field
@@ -37,8 +36,7 @@ class RegressionInput(BaseModel):
 
 def _log_sklearn_model(model, tmp_path):
     import mlflow.sklearn
-    db_uri = f"sqlite:///{tmp_path}/mlflow.db"
-    mlflow.set_tracking_uri(db_uri)
+    mlflow.set_tracking_uri(f"sqlite:///{tmp_path}/mlflow.db")
     mlflow.set_experiment("predikit-test")
     with mlflow.start_run() as run:
         mlflow.sklearn.log_model(model, artifact_path="model")