uncertainty-engine-types 0.0.11__py3-none-any.whl → 0.0.13__py3-none-any.whl

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/job.py

@@ -7,6 +7,7 @@ from pydantic import BaseModel
 class JobStatus(Enum):
     PENDING = "pending"
     RUNNING = "running"
+    CANCELLED = "cancelled"
     COMPLETED = "completed"
     FAILED = "failed"
 

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/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/version.py

@@ -1 +1 @@
-__version__ = "0.0.11"
+__version__ = "0.0.13"

{uncertainty_engine_types-0.0.11.dist-info → uncertainty_engine_types-0.0.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: uncertainty-engine-types
-Version: 0.0.11
+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.11.dist-info → uncertainty_engine_types-0.0.13.dist-info}/RECORD

@@ -1,4 +1,4 @@
-uncertainty_engine_types/__init__.py,sha256=_C9BE7MJkVmE1Ky2AVhSrcAro-QPf7dsNsbQTUcxZJc,1797
+uncertainty_engine_types/__init__.py,sha256=yUPp9RpvuwNXZUpmVwcqZ8Q51Oa8d9j0ika5sw0weB0,1866
 uncertainty_engine_types/chat_history.py,sha256=OY1fZXP7_AtPKEmgjxh60PlbEGtQWJWafZMgs2Ec0BU,121
 uncertainty_engine_types/context.py,sha256=k9rlArn-L2x1P5vWy5khfuqJW5tJOqr3NlhubZCbyG4,209
 uncertainty_engine_types/dataset.py,sha256=sDpQu5X3KxJ1lNkZOwhppJ2SY0Cv19cbcYfGpp_hyUQ,75
@@ -8,19 +8,20 @@ uncertainty_engine_types/file.py,sha256=J202bhOIVnvvvAG54sXyYF8fwjZGTUP3097ARnRn
 uncertainty_engine_types/graph.py,sha256=ii2YT2jxoRB7UhcceI2k_ArKmlUXEwBCRBpsEUjL2Qg,265
 uncertainty_engine_types/handle.py,sha256=buX_xEdBwT7vZRs5uCEPB6VCbSWIFbCE-NYIL4fZSjg,942
 uncertainty_engine_types/id.py,sha256=Wpco2lLEsx6sdfdRJcPGl8GQlVa13dflWApaRt6dTYI,74
-uncertainty_engine_types/job.py,sha256=C3dS2NWL3-Nw2jb5x5VqGNMPWuYcKhIc219K4cNpBMw,341
+uncertainty_engine_types/job.py,sha256=J_7tlWufZprOlePIEfBmir-5ZY9YXHmmALK_Ykj0ubs,369
 uncertainty_engine_types/llm.py,sha256=Ae7dw1R5RKLJHFjNwBzynLd-ddikKP0NsvHvoMdmnLM,916
 uncertainty_engine_types/message.py,sha256=5IY-S6Ipt6oYmnVSkRVv3QLSH-lKQvpxp_qF5YGSbok,209
 uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk-y68,196
-uncertainty_engine_types/model_config.py,sha256=Ghii7gk3zRhydGQJJJlzYRmMFW2cHBfwGj9FPNHEhCQ,320
+uncertainty_engine_types/model_config.py,sha256=AeN5oR06uZJMGoAJoaI_Ix4nCy5nNcU2Wh69ijP39rQ,286
 uncertainty_engine_types/node_info.py,sha256=2p87fCdX_CrdT2ZODtGJ-mR1Mvg46WhaE1GMddaEVak,800
 uncertainty_engine_types/prompt.py,sha256=l__qXytAapKg1Hoaj3RSVYN3rhy638nuUZIS-se0eMc,74
 uncertainty_engine_types/sensor_designer.py,sha256=hr3ek4_dRjRK0-78uaT6h8-bGUCm7Mfs6mxJSWxE64c,80
 uncertainty_engine_types/sql.py,sha256=SBzmgMEZ-sa-OBRoZbmKiOqddop4zJixgxx-JCus9fY,298
 uncertainty_engine_types/token.py,sha256=4tQQkvl-zsYoVk8ZEx0cB2JiU0VDRJ6uUe76XBXEpBY,178
+uncertainty_engine_types/uncertainty_plot.py,sha256=kJr0SuJ6JeTxaf2adpDNWqx7vVLZRpKG8tFbdx90pas,547
 uncertainty_engine_types/utils.py,sha256=72QVig8Kb5uIR-e1nofm-3x9CouebdQJIruDbq-aIn0,271
 uncertainty_engine_types/vector_store.py,sha256=9fYPJ04jWcy2DruyUSjiKQAgmqq-wgeAi5dBIrAOm30,392
-uncertainty_engine_types/version.py,sha256=OaIl7v-6zEWEY90Jlh6yoBjO3zWX1oX7RsvqrK3o1TU,23
-uncertainty_engine_types-0.0.11.dist-info/METADATA,sha256=6VC6b9iAOQDiRv0NO9FP2RGl59TbWMnSiY5zob1TDic,2816
-uncertainty_engine_types-0.0.11.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-uncertainty_engine_types-0.0.11.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=SxvRes_80c-42zwvn5BXPvpGVs_EZ69FUEtUBF1k9Ts,23
+uncertainty_engine_types-0.0.13.dist-info/METADATA,sha256=lmGp8vQgY44E6PoGzr8CfFz0sjkw05L4ZQZ7yWmPSJw,2816
+uncertainty_engine_types-0.0.13.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+uncertainty_engine_types-0.0.13.dist-info/RECORD,,