uncertainty-engine-types 0.0.12__tar.gz → 0.0.13__tar.gz

{uncertainty_engine_types-0.0.12 → uncertainty_engine_types-0.0.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.0.12
+Version: 0.0.13
 Summary: Common type definitions for the Uncertainty Engine
 Author: Freddy Wordingham
 Author-email: freddy@digilab.ai

{uncertainty_engine_types-0.0.12 → uncertainty_engine_types-0.0.13}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uncertainty-engine-types"
-version = "0.0.12"
+version = "0.0.13"
 description = "Common type definitions for the Uncertainty Engine"
 authors = [
     { name = "Freddy Wordingham", email = "freddy@digilab.ai" },

{uncertainty_engine_types-0.0.12 → uncertainty_engine_types-0.0.13}/uncertainty_engine_types/__init__.py

@@ -30,6 +30,7 @@ from .prompt import Prompt
 from .sensor_designer import SensorDesigner
 from .sql import SQLConfig, SQLKind
 from .token import Token
+from .uncertainty_plot import UncertaintyPlot
 from .vector_store import VectorStoreConfig, VectorStoreProvider
 from .version import __version__
 
@@ -73,6 +74,7 @@ __all__ = [
     "TextEmbeddingsConfig",
     "TextEmbeddingsProvider",
     "Token",
+    "UncertaintyPlot",
     "utils",
     "VectorStoreConfig",
     "VectorStoreProvider",

{uncertainty_engine_types-0.0.12 → uncertainty_engine_types-0.0.13}/uncertainty_engine_types/model_config.py

@@ -4,7 +4,6 @@ from pydantic import BaseModel
 
 
 class ModelConfig(BaseModel):
-    train_test_ratio: float = 1.0
     input_variance: Optional[float] = None
     output_variance: Optional[float] = None
     model_type: Literal["SingleTaskGPTorch"] = "SingleTaskGPTorch"

uncertainty_engine_types-0.0.13/uncertainty_engine_types/uncertainty_plot.py

@@ -0,0 +1,19 @@
+from pydantic import BaseModel, model_validator
+
+
+class UncertaintyPlot(BaseModel, extra="forbid"):
+    x_label: str
+    y_label: str
+    x_vals: list[float]
+    mean: list[float]
+    std: list[float]
+    lower: list[float]
+    upper: list[float]
+
+    @model_validator(mode="after")
+    def validate_length(cls, model):
+        n = len(model.x_vals)
+        for _list in [model.mean, model.std, model.lower, model.upper]:
+            if n != len(_list):
+                raise ValueError("All lists must have the same length")
+        return model

uncertainty_engine_types-0.0.13/uncertainty_engine_types/version.py

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

uncertainty_engine_types-0.0.12/uncertainty_engine_types/version.py

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