predikit 0.3.1__tar.gz → 0.3.3__tar.gz

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

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: predikit
-Version: 0.3.1
+Version: 0.3.3
 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
 Project-URL: Bug Tracker, https://github.com/Tejas-TA/predikit/issues
 Project-URL: Changelog, https://github.com/Tejas-TA/predikit/releases
-Author-email: Tejas Tumakuru Ashok <tejas.tumakuruashok@travelandleisure.com>
+Author-email: Tejas Tumakuru Ashok <tejasta@gmail.com>
 License: MIT
 License-File: LICENSE
 Keywords: agents,function-calling,llm,ml-tools,sklearn,xgboost
@@ -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.3}/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.3}/pyproject.toml

@@ -4,13 +4,13 @@ build-backend = "hatchling.build"
 
 [project]
 name = "predikit"
-version = "0.3.1"
+version = "0.3.3"
 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"}
 requires-python = ">=3.10"
 authors = [
-    {name = "Tejas Tumakuru Ashok", email = "tejas.tumakuruashok@travelandleisure.com"},
+    {name = "Tejas Tumakuru Ashok", email = "tejasta@gmail.com"},
 ]
 keywords = ["llm", "agents", "sklearn", "xgboost", "function-calling", "ml-tools"]
 classifiers = [

{predikit-0.3.1 → predikit-0.3.3}/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.3"

{predikit-0.3.1 → predikit-0.3.3}/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")