uncertainty-engine-types 0.0.12__py3-none-any.whl → 0.0.14__py3-none-any.whl

uncertainty_engine_types/__init__.py

@@ -1,6 +1,6 @@
 from . import utils
 from .chat_history import ChatHistory
-from .context import Context
+from .context import Context, UserContext
 from .dataset import CSVDataset
 from .embeddings import TextEmbeddingsConfig, TextEmbeddingsProvider
 from .execution_error import ExecutionError
@@ -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__
 
@@ -59,6 +60,7 @@ __all__ = [
     "NodeInfo",
     "NodeInputInfo",
     "NodeOutputInfo",
+    "NodeRequirementsInfo",
     "PDF",
     "Prompt",
     "ResourceID",
@@ -73,6 +75,8 @@ __all__ = [
     "TextEmbeddingsConfig",
     "TextEmbeddingsProvider",
     "Token",
+    "UserContext",
+    "UncertaintyPlot",
     "utils",
     "VectorStoreConfig",
     "VectorStoreProvider",

uncertainty_engine_types/context.py

@@ -3,6 +3,12 @@ from pydantic import BaseModel
 from .node_info import NodeInfo
 
 
+class UserContext(BaseModel):
+    email: str
+    project_id: str
+    cost_code: str
+
+
 class Context(BaseModel):
     sync: bool
     job_id: str
@@ -10,3 +16,4 @@ class Context(BaseModel):
     cache_url: str
     timeout: int
     nodes: dict[str, NodeInfo]
+    user: UserContext

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

@@ -20,7 +20,24 @@ class NodeOutputInfo(BaseModel):
     description: str
 
 
-class NodeInfo(BaseModel):
+class NodeRequirementsInfo(BaseModel):
+    cpu: int
+    gpu: bool
+    memory: int
+    timeout: int
+
+
+class NodeInfo(BaseModel, extra="allow"):
+    """
+    Node information.
+    """
+
+    # New properties must be added as optional. The Node Registry uses this
+    # model and must support Nodes that don't provide a full set of details.
+    #
+    # Likewise, the `extra="allow"` argument allows the Node Registry to
+    # deserialise `NodeInfo` models with properties added post-release.
+
     id: str
     label: str
     category: str
@@ -30,6 +47,11 @@ class NodeInfo(BaseModel):
     cost: int
     inputs: dict[str, NodeInputInfo]
     outputs: dict[str, NodeOutputInfo] = {}
+    requirements: Optional[NodeRequirementsInfo] = None
+    """
+    Deployment requirements.
+    """
+
     load_balancer_url: Optional[str] = None
     queue_url: Optional[str] = None
     cache_url: Optional[str] = None

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.12"
+__version__ = "0.0.14"

{uncertainty_engine_types-0.0.12.dist-info → uncertainty_engine_types-0.0.14.dist-info}/METADATA

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

{uncertainty_engine_types-0.0.12.dist-info → uncertainty_engine_types-0.0.14.dist-info}/RECORD

@@ -1,6 +1,6 @@
-uncertainty_engine_types/__init__.py,sha256=_C9BE7MJkVmE1Ky2AVhSrcAro-QPf7dsNsbQTUcxZJc,1797
+uncertainty_engine_types/__init__.py,sha256=jK69AKVZ6BkdrH0zGHPfLdTU9BPVad_AwJiKAr1Tr30,1926
 uncertainty_engine_types/chat_history.py,sha256=OY1fZXP7_AtPKEmgjxh60PlbEGtQWJWafZMgs2Ec0BU,121
-uncertainty_engine_types/context.py,sha256=k9rlArn-L2x1P5vWy5khfuqJW5tJOqr3NlhubZCbyG4,209
+uncertainty_engine_types/context.py,sha256=1iDcfMLi_fKh_YVUAKif2NOdrA_vuXOQvBZKbF4uJZk,317
 uncertainty_engine_types/dataset.py,sha256=sDpQu5X3KxJ1lNkZOwhppJ2SY0Cv19cbcYfGpp_hyUQ,75
 uncertainty_engine_types/embeddings.py,sha256=sNDWjpuqmzOefKyQX--9EpdeTEyI4kQeSH4IELVUT7g,982
 uncertainty_engine_types/execution_error.py,sha256=tvIBuZPM8UhFUERHCEqoW8blAYYuBq1ZqidO0i_BCGs,105
@@ -12,15 +12,16 @@ uncertainty_engine_types/job.py,sha256=J_7tlWufZprOlePIEfBmir-5ZY9YXHmmALK_Ykj0u
 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/node_info.py,sha256=2p87fCdX_CrdT2ZODtGJ-mR1Mvg46WhaE1GMddaEVak,800
+uncertainty_engine_types/model_config.py,sha256=AeN5oR06uZJMGoAJoaI_Ix4nCy5nNcU2Wh69ijP39rQ,286
+uncertainty_engine_types/node_info.py,sha256=XHONP6rNbAcLOHFxi-NPl1-UrKAH9P8C1yUr_rDiyCg,1362
 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=1SFj-qMOaeO0fOKQaFmYhAKiRwwNda5uWKpYtIcken8,23
-uncertainty_engine_types-0.0.12.dist-info/METADATA,sha256=ek29yMi9iXJbKzn-5SlUpo5XNeewelcu_Cc7B0T2Mzk,2816
-uncertainty_engine_types-0.0.12.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-uncertainty_engine_types-0.0.12.dist-info/RECORD,,
+uncertainty_engine_types/version.py,sha256=71Tw8stu19MsLPyF1q_DF5mb_BefzOTPYggLlSLvCms,23
+uncertainty_engine_types-0.0.14.dist-info/METADATA,sha256=DKy3FpsiiBed4unRmOVzEi2e6XW352y8Iip5KLApHKE,2816
+uncertainty_engine_types-0.0.14.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+uncertainty_engine_types-0.0.14.dist-info/RECORD,,