uncertainty-engine-types 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. uncertainty_engine_types/job.py +3 -0
  2. uncertainty_engine_types/model_config.py +7 -1
  3. uncertainty_engine_types/version.py +1 -1
  4. {uncertainty_engine_types-0.5.0.dist-info → uncertainty_engine_types-0.7.0.dist-info}/METADATA +1 -1
  5. {uncertainty_engine_types-0.5.0.dist-info → uncertainty_engine_types-0.7.0.dist-info}/RECORD +6 -6
  6. {uncertainty_engine_types-0.5.0.dist-info → uncertainty_engine_types-0.7.0.dist-info}/WHEEL +1 -1
uncertainty_engine_types/job.py CHANGED
@@ -11,6 +11,9 @@ class JobStatus(Enum):
11
11
  COMPLETED = "completed"
12
12
  FAILED = "failed"
13
13
 
14
+ def is_terminal(self) -> bool:
15
+ return self in [JobStatus.COMPLETED, JobStatus.FAILED]
16
+
14
17
 
15
18
  class JobInfo(BaseModel):
16
19
  status: JobStatus
uncertainty_engine_types/model_config.py CHANGED
@@ -2,13 +2,19 @@ from typing import Literal, Optional
2
2
 
3
3
  from pydantic import BaseModel
4
4
 
5
+ ModelType = Literal[
6
+ "BernoulliClassificationGPTorch",
7
+ "SingleTaskGPTorch",
8
+ "SingleTaskVariationalGPTorch",
9
+ ]
10
+
5
11
 
6
12
  class ModelConfig(BaseModel):
7
13
  input_variance: Optional[float] = None
8
14
  input_retained_dimensions: Optional[int] = None
9
15
  output_variance: Optional[float] = None
10
16
  output_retained_dimensions: Optional[int] = None
11
- model_type: Literal["SingleTaskGPTorch"] = "SingleTaskGPTorch"
17
+ model_type: ModelType = "SingleTaskGPTorch"
12
18
  kernel: Optional[str] = None
13
19
  warp_inputs: bool = False
14
20
  seed: Optional[int] = None
uncertainty_engine_types/version.py CHANGED
@@ -1 +1 @@
1
- __version__ = "0.5.0"
1
+ __version__ = "0.7.0"
{uncertainty_engine_types-0.5.0.dist-info → uncertainty_engine_types-0.7.0.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: uncertainty-engine-types
3
- Version: 0.5.0
3
+ Version: 0.7.0
4
4
  Summary: Common type definitions for the Uncertainty Engine
5
5
  Author: Freddy Wordingham
6
6
  Author-email: freddy@digilab.ai
{uncertainty_engine_types-0.5.0.dist-info → uncertainty_engine_types-0.7.0.dist-info}/RECORD RENAMED
@@ -9,11 +9,11 @@ uncertainty_engine_types/file.py,sha256=J202bhOIVnvvvAG54sXyYF8fwjZGTUP3097ARnRn
9
9
  uncertainty_engine_types/graph.py,sha256=ii2YT2jxoRB7UhcceI2k_ArKmlUXEwBCRBpsEUjL2Qg,265
10
10
  uncertainty_engine_types/handle.py,sha256=buX_xEdBwT7vZRs5uCEPB6VCbSWIFbCE-NYIL4fZSjg,942
11
11
  uncertainty_engine_types/id.py,sha256=Wpco2lLEsx6sdfdRJcPGl8GQlVa13dflWApaRt6dTYI,74
12
- uncertainty_engine_types/job.py,sha256=mKNeZ9Ay7SdO7VH9kYS74U7ptcrmQIXyrGVgiE7MRz8,424
12
+ uncertainty_engine_types/job.py,sha256=y5ko9XYGDJ5SOJSXfompFPCmnQplJLl5LqHpK-zPVWo,523
13
13
  uncertainty_engine_types/llm.py,sha256=Ae7dw1R5RKLJHFjNwBzynLd-ddikKP0NsvHvoMdmnLM,916
14
14
  uncertainty_engine_types/message.py,sha256=sX1tq_BbZC9r9-Vf0A3WHiOFRE2hAAlY-7rD52kbsoU,381
15
15
  uncertainty_engine_types/model.py,sha256=O9E_7DE9AKEc1o2VnhpUyl3Quh4sGdV43gqDJwk-y68,196
16
- uncertainty_engine_types/model_config.py,sha256=JG9PXbHaMPhEpdrhTSK8S0LpyKzUUrDEyx1K4bKBSDQ,454
16
+ uncertainty_engine_types/model_config.py,sha256=pCawKIo2l0O3eRLE5Ese7jaxomIyPXGW3XIcAx6TckY,558
17
17
  uncertainty_engine_types/node_info.py,sha256=XHONP6rNbAcLOHFxi-NPl1-UrKAH9P8C1yUr_rDiyCg,1362
18
18
  uncertainty_engine_types/prompt.py,sha256=l__qXytAapKg1Hoaj3RSVYN3rhy638nuUZIS-se0eMc,74
19
19
  uncertainty_engine_types/sensor_designer.py,sha256=hr3ek4_dRjRK0-78uaT6h8-bGUCm7Mfs6mxJSWxE64c,80
@@ -22,7 +22,7 @@ uncertainty_engine_types/token.py,sha256=4tQQkvl-zsYoVk8ZEx0cB2JiU0VDRJ6uUe76XBX
22
22
  uncertainty_engine_types/uncertainty_plot.py,sha256=kJr0SuJ6JeTxaf2adpDNWqx7vVLZRpKG8tFbdx90pas,547
23
23
  uncertainty_engine_types/utils.py,sha256=72QVig8Kb5uIR-e1nofm-3x9CouebdQJIruDbq-aIn0,271
24
24
  uncertainty_engine_types/vector_store.py,sha256=9fYPJ04jWcy2DruyUSjiKQAgmqq-wgeAi5dBIrAOm30,392
25
- uncertainty_engine_types/version.py,sha256=LBK46heutvn3KmsCrKIYu8RQikbfnjZaj2xFrXaeCzQ,22
26
- uncertainty_engine_types-0.5.0.dist-info/METADATA,sha256=QQPePJ01ZHO5ThDVRE2Di3cvymS08oijLw6qCLc6VUs,2815
27
- uncertainty_engine_types-0.5.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
28
- uncertainty_engine_types-0.5.0.dist-info/RECORD,,
25
+ uncertainty_engine_types/version.py,sha256=RaANGbRu5e-vehwXI1-Qe2ggPPfs1TQaZj072JdbLk4,22
26
+ uncertainty_engine_types-0.7.0.dist-info/METADATA,sha256=0t50EYJpW4a2LeueibBBfQcX2hpJ0r4t2he5TMkVAFg,2815
27
+ uncertainty_engine_types-0.7.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
28
+ uncertainty_engine_types-0.7.0.dist-info/RECORD,,
{uncertainty_engine_types-0.5.0.dist-info → uncertainty_engine_types-0.7.0.dist-info}/WHEEL RENAMED
@@ -1,4 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: poetry-core 2.1.1
2
+ Generator: poetry-core 2.1.2
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any