validmind 2.14.0__tar.gz → 2.15.0__tar.gz

{validmind-2.14.0 → validmind-2.15.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: validmind
-Version: 2.14.0
+Version: 2.15.0
 Summary: ValidMind Library
 Author-email: Andres Rodriguez <andres@validmind.ai>, Juan Martinez <juan@validmind.ai>, Anil Sorathiya <anil@validmind.ai>, Luis Pallares <luis@validmind.ai>, John Walz <john@validmind.ai>
 License:                          DUAL LICENSE NOTICE
@@ -696,6 +696,7 @@ Requires-Dist: aiohttp[speedups]<3.13.1
 Requires-Dist: anywidget
 Requires-Dist: beautifulsoup4
 Requires-Dist: ipywidgets
+Requires-Dist: jinja2>=3.1.2
 Requires-Dist: kaleido<2.0.0,>=1.2.0
 Requires-Dist: matplotlib
 Requires-Dist: mistune<4.0.0,>=3.0.2

{validmind-2.14.0 → validmind-2.15.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "validmind"
-version = "2.14.0"
+version = "2.15.0"
 description = "ValidMind Library"
 readme = "README.pypi.md"
 requires-python = ">=3.9,<3.15"
@@ -16,6 +16,7 @@ dependencies = [
   "aiohttp[speedups] (<3.13.1)",
   "requests (>=2.28.0,<3.0.0)",
   "ipywidgets",
+  "jinja2 (>=3.1.2)",
   "kaleido (>=1.2.0,<2.0.0)",
   "matplotlib",
   "mistune (>=3.0.2,<4.0.0)",
@@ -198,3 +199,14 @@ exclude-newer = "7 days"
 exclude-newer-package = { certifi = "2026-10-21" }
 # Security floor: certifi>=2026.10.19 distrusts compromised eMudhra intermediates (Mozilla bug 2083449)
 constraint-dependencies = ["certifi>=2026.10.19"]
+
+# CPU-only torch for this repo's own installs (dev, CI). CI runners have no GPU, and PyPI's Linux
+# torch pulls ~3 GB of NVIDIA libraries. uv-only setting: the published package still depends on
+# plain `torch`, so what users install is unchanged.
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
+[tool.uv.sources]
+torch = [{ index = "pytorch-cpu", marker = "sys_platform == 'linux'" }]

validmind-2.15.0/validmind/__version__.py

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

{validmind-2.14.0 → validmind-2.15.0}/validmind/ai/test_descriptions.py

@@ -220,6 +220,14 @@ def background_generate_description(
 
             return test_description, False
 
+    # Render figures to PNG here, on the calling thread: Plotly's renderer (kaleido)
+    # deadlocks when several description workers start it at the same time.
+    for figure in figures or []:
+        try:
+            figure.pre_serialize()
+        except Exception:
+            pass  # the worker hits the same error and falls back to the default
+
     return DescriptionFuture(__executor.submit(wrapped))
 
 

{validmind-2.14.0 → validmind-2.15.0}/validmind/api_client.py

@@ -50,6 +50,7 @@ _oidc_expires_at: Optional[str] = None
 _oidc_refresh_lock = threading.Lock()
 
 __api_session: Optional[aiohttp.ClientSession] = None
+__api_session_loop: Optional[asyncio.AbstractEventLoop] = None
 
 
 def _invalidate_async_session() -> None:
@@ -72,23 +73,10 @@ def _invalidate_async_session() -> None:
 @atexit.register
 def _close_session():
     """Closes the async client session at exit."""
-    if __api_session and not __api_session.closed:
-        try:
-            loop = asyncio.get_event_loop()
-            if loop.is_running():
-                loop.create_task(__api_session.close())
-            else:
-                loop.run_until_complete(__api_session.close())
-        except RuntimeError as e:
-            # ignore RuntimeError when closing the session from the main thread
-            if "no current event loop in thread" in str(e):
-                pass
-            elif "Event loop is closed" in str(e):
-                pass
-            else:
-                raise e
-        except Exception as e:
-            logger.exception("Error closing aiohttp session at exit: %s", e)
+    try:
+        _invalidate_async_session()
+    except Exception as e:
+        logger.exception("Error closing aiohttp session at exit: %s", e)
 
 
 def get_api_host() -> Optional[str]:
@@ -120,8 +108,14 @@ def _get_api_headers() -> Dict[str, str]:
 
 
 def _get_session() -> aiohttp.ClientSession:
-    """Initializes the async client session."""
-    global __api_session
+    """Initializes the async client session for the running event loop."""
+    global __api_session, __api_session_loop
+
+    # An aiohttp session only works on the loop it was created on, and outside a
+    # notebook run_async starts a new loop (asyncio.run) for every call.
+    loop = asyncio.get_running_loop()
+    if __api_session_loop is not loop:
+        _invalidate_async_session()
 
     if not __api_session or __api_session.closed:
         __api_session = aiohttp.ClientSession(
@@ -129,6 +123,7 @@ def _get_session() -> aiohttp.ClientSession:
             timeout=aiohttp.ClientTimeout(total=int(os.getenv("VM_API_TIMEOUT", 30))),
             trust_env=True,
         )
+        __api_session_loop = loop
 
     return __api_session
 

{validmind-2.14.0 → validmind-2.15.0}/validmind/datasets/regression/__init__.py

@@ -10,6 +10,8 @@ from typing import List
 import pandas as pd
 
 __all__: List[str] = [
+    "california_housing",
+    "credit_card_balance",
     "fred",
     "lending_club",
 ]

validmind-2.15.0/validmind/datasets/regression/credit_card_balance.py

@@ -0,0 +1,57 @@
+# Copyright © 2023-2026 ValidMind Inc. All rights reserved.
+# Refer to the LICENSE file in the root of this repository for details.
+# SPDX-License-Identifier: AGPL-3.0 AND ValidMind Commercial
+
+"""
+Synthetic credit card balance dataset for linear regression.
+
+Each row is a cardholder observed in one quarter; the target is their average monthly
+revolving balance. The data is synthetic — see `generate_credit_card_balance_csv.py`
+for how it was made.
+"""
+
+import os
+
+import pandas as pd
+from sklearn.model_selection import train_test_split
+
+current_path = os.path.dirname(os.path.abspath(__file__))
+dataset_path = os.path.join(current_path, "datasets")
+
+feature_columns = [
+    "Income",
+    "DebtToIncome",
+    "CreditScore",
+    "CreditLimit",
+    "RecentInquiries",
+    "NumCards",
+    "TenureYears",
+    "IsStudent",
+    "Region",
+]
+target_column = "Balance"
+time_column = "Quarter"
+oot_start = "2025Q3"
+
+bundled_data_file = os.path.join(dataset_path, "credit_card_balance.csv")
+
+
+def load_data():
+    """Load the bundled synthetic credit card balance dataset."""
+    return pd.read_csv(bundled_data_file)
+
+
+def preprocess(df, oot_start=oot_start, test_size=0.2, random_state=42):
+    """
+    Split into train and test sets from the development period, and an out-of-time
+    set from `oot_start` onwards.
+
+    Returns:
+        tuple: (train_df, test_df, oot_df)
+    """
+    development = df[df[time_column] < oot_start]
+    oot_df = df[df[time_column] >= oot_start].copy()
+    train_df, test_df = train_test_split(
+        development.copy(), test_size=test_size, random_state=random_state
+    )
+    return train_df, test_df, oot_df