validmind 2.7.5__tar.gz → 2.7.7__tar.gz

{validmind-2.7.5 → validmind-2.7.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: validmind
-Version: 2.7.5
+Version: 2.7.7
 Summary: ValidMind Library
 License: Commercial License
 Author: Andres Rodriguez
@@ -26,7 +26,6 @@ Requires-Dist: ipywidgets
 Requires-Dist: kaleido (>=0.2.1,!=0.2.1.post1)
 Requires-Dist: langchain-openai (>=0.1.8) ; extra == "all" or extra == "llm"
 Requires-Dist: langdetect
-Requires-Dist: latex2mathml (>=3.77.0)
 Requires-Dist: llvmlite ; python_version >= "3.8" and python_full_version <= "3.11.0"
 Requires-Dist: matplotlib
 Requires-Dist: mistune (>=3.0.2,<4.0.0)
@@ -44,7 +43,7 @@ Requires-Dist: python-dotenv
 Requires-Dist: ragas (>=0.2.3) ; extra == "all" or extra == "llm"
 Requires-Dist: rouge (>=1)
 Requires-Dist: rpy2 (>=3.5.10,<4.0.0) ; extra == "all" or extra == "r-support"
-Requires-Dist: scikit-learn (!=1.6.0)
+Requires-Dist: scikit-learn (<1.6.0)
 Requires-Dist: scipy
 Requires-Dist: scorecardpy (>=0.1.9.6,<0.2.0.0)
 Requires-Dist: seaborn

{validmind-2.7.5 → validmind-2.7.7}/pyproject.toml

@@ -10,7 +10,7 @@ description = "ValidMind Library"
 license = "Commercial License"
 name = "validmind"
 readme = "README.pypi.md"
-version = "2.7.5"
+version = "2.7.7"
 
 [tool.poetry.dependencies]
 aiohttp = {extras = ["speedups"], version = "*"}
@@ -23,7 +23,6 @@ ipywidgets = "*"
 kaleido = ">=0.2.1,!=0.2.1.post1"
 langchain-openai = {version = ">=0.1.8", optional = true}
 langdetect = "*"
-latex2mathml = ">=3.77.0"
 llvmlite = {version = "*", python = ">=3.8,<=3.11"}
 matplotlib = "*"
 mistune = "^3.0.2"
@@ -42,7 +41,7 @@ python-dotenv = "*"
 ragas = {version = ">=0.2.3", optional = true}
 rouge = ">=1"
 rpy2 = {version = "^3.5.10", optional = true}
-scikit-learn = "*,!=1.6.0"
+scikit-learn = "*,<1.6.0"
 scipy = "*"
 scorecardpy = "^0.1.9.6"
 seaborn = "*"

{validmind-2.7.5 → validmind-2.7.7}/validmind/__init__.py

@@ -50,6 +50,7 @@ from .client import (  # noqa: E402
     run_test_suite,
 )
 from .tests.decorator import tags, tasks, test
+from .tests.run import print_env
 from .vm_models.result import RawData
 
 __all__ = [  # noqa
@@ -63,6 +64,7 @@ __all__ = [  # noqa
     "init_model",
     "init_r_model",
     "preview_template",
+    "print_env",
     "RawData",
     "reload",
     "run_documentation_tests",

validmind-2.7.7/validmind/__version__.py

@@ -0,0 +1 @@
+__version__ = "2.7.7"

{validmind-2.7.5 → validmind-2.7.7}/validmind/api_client.py

@@ -407,6 +407,7 @@ async def alog_metric(
     inputs: Optional[List[str]] = None,
     params: Optional[Dict[str, Any]] = None,
     recorded_at: Optional[str] = None,
+    thresholds: Optional[Dict[str, Any]] = None,
 ):
     """See log_metric for details"""
     if not key or not isinstance(key, str):
@@ -421,6 +422,9 @@ async def alog_metric(
         except (ValueError, TypeError):
             raise ValueError("`value` must be a scalar (int or float)")
 
+    if thresholds is not None and not isinstance(thresholds, dict):
+        raise ValueError("`thresholds` must be a dictionary or None")
+
     try:
         return await _post(
             "log_unit_metric",
@@ -431,6 +435,7 @@ async def alog_metric(
                     "inputs": inputs or [],
                     "params": params or {},
                     "recorded_at": recorded_at,
+                    "thresholds": thresholds or {},
                 },
                 cls=NumpyEncoder,
                 allow_nan=False,
@@ -447,6 +452,7 @@ def log_metric(
     inputs: Optional[List[str]] = None,
     params: Optional[Dict[str, Any]] = None,
     recorded_at: Optional[str] = None,
+    thresholds: Optional[Dict[str, Any]] = None,
 ):
     """Logs a unit metric
 
@@ -463,8 +469,9 @@ def log_metric(
         params (dict, optional): Dictionary of parameters used to compute the metric.
         recorded_at (str, optional): The timestamp of the metric. Server will use
             current time if not provided.
+        thresholds (dict, optional): Dictionary of thresholds for the metric.
     """
-    run_async(alog_metric, key, value, inputs, params, recorded_at)
+    run_async(alog_metric, key, value, inputs, params, recorded_at, thresholds)
 
 
 def get_ai_key() -> Dict[str, Any]: