dyff-schema 0.38.1__py3-none-any.whl → 0.38.3__py3-none-any.whl

dyff/schema/_version.py

@@ -1,2 +1,2 @@
-__version__ = version = "0.38.1"
-__version_tuple__ = version_tuple = (0, 38, 1)
+__version__ = version = "0.38.3"
+__version_tuple__ = version_tuple = (0, 38, 3)

dyff/schema/v0/r1/platform.py

@@ -2832,6 +2832,7 @@ class PipelineBase(DyffSchemaBaseModel):
         for k, v in nodes.items():
             if k != v.name:
                 raise ValueError(f"nodes[{k}]: dict key must match value.name")
+        return nodes
 
 
 class Pipeline(DyffEntity, PipelineBase):
@@ -2883,6 +2884,12 @@ class PipelineRun(DyffEntity, PipelineRunBase):
         return None
 
 
+class PipelineRunStatus(Status):
+    nodes: dict[str, Status] = pydantic.Field(
+        default_factory=dict, description="The status of each pipeline node."
+    )
+
+
 # ---------------------------------------------------------------------------
 # Status enumerations
 
@@ -3123,34 +3130,6 @@ def is_status_success(status: str) -> bool:
     return status in [EntityStatus.complete, EntityStatus.ready]
 
 
-_ENTITY_CLASS = {
-    Entities.Analysis: Analysis,
-    Entities.Artifact: OCIArtifact,
-    Entities.Audit: Audit,
-    Entities.AuditProcedure: AuditProcedure,
-    Entities.Challenge: Challenge,
-    Entities.Dataset: Dataset,
-    Entities.DataSource: DataSource,
-    Entities.Evaluation: Evaluation,
-    Entities.Family: Family,
-    Entities.Hazard: Hazard,
-    Entities.InferenceService: InferenceService,
-    Entities.InferenceSession: InferenceSession,
-    Entities.Measurement: Measurement,
-    Entities.Method: Method,
-    Entities.Model: Model,
-    Entities.Module: Module,
-    Entities.Report: Report,
-    Entities.SafetyCase: SafetyCase,
-    Entities.Team: Team,
-    Entities.UseCase: UseCase,
-}
-
-
-def entity_class(kind: Entities):
-    return _ENTITY_CLASS[kind]
-
-
 # A Union type containing all Dyff top-level schema types, except the ones that
 # should not have Revisions generated for them when they change
 _DyffEntityTypeRevisable = Union[
@@ -3230,6 +3209,37 @@ class Revision(DyffEntity, RevisionMetadata):
 DyffEntityType = Union[_DyffEntityTypeRevisable, History, Revision]
 
 
+_ENTITY_CLASS: dict[Entities, type[DyffEntityType]] = {
+    Entities.Artifact: OCIArtifact,
+    Entities.Audit: Audit,
+    Entities.AuditProcedure: AuditProcedure,
+    Entities.Challenge: Challenge,
+    Entities.Dataset: Dataset,
+    Entities.DataSource: DataSource,
+    Entities.Evaluation: Evaluation,
+    Entities.Family: Family,
+    Entities.Hazard: Hazard,
+    Entities.History: History,
+    Entities.InferenceService: InferenceService,
+    Entities.InferenceSession: InferenceSession,
+    Entities.Measurement: Measurement,
+    Entities.Method: Method,
+    Entities.Model: Model,
+    Entities.Module: Module,
+    Entities.Pipeline: Pipeline,
+    Entities.PipelineRun: PipelineRun,
+    Entities.Report: Report,
+    Entities.Revision: Revision,
+    Entities.SafetyCase: SafetyCase,
+    Entities.Team: Team,
+    Entities.UseCase: UseCase,
+}
+
+
+def entity_class(kind: Entities) -> type[DyffEntityType]:
+    return _ENTITY_CLASS[kind]
+
+
 # A TypeVar that matches any Dyff top-level schema type
 DyffEntityT = TypeVar(
     "DyffEntityT",
@@ -3401,6 +3411,7 @@ __all__ = [
     "PipelineParameter",
     "PipelineRun",
     "PipelineRunBase",
+    "PipelineRunStatus",
     "PipelineSafetyCaseRequest",
     "QueryableDyffEntity",
     "Report",

{dyff_schema-0.38.1.dist-info → dyff_schema-0.38.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dyff-schema
-Version: 0.38.1
+Version: 0.38.3
 Summary: Data models for the Dyff AI auditing platform.
 Author-email: Digital Safety Research Institute <contact@dsri.org>
 License: Apache-2.0

{dyff_schema-0.38.1.dist-info → dyff_schema-0.38.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 dyff/schema/__init__.py,sha256=w7OWDFuyGKd6xt_yllNtKzHahPgywrfU4Ue02psYaMA,2244
-dyff/schema/_version.py,sha256=OAbKv7VQqg-KzyqDjyeWStot6XPBinPJdHJGJ1MwAw4,80
+dyff/schema/_version.py,sha256=-CCWFr62vN0w7SrCBH-IeTHMoX8ScRHffpLPQafa5pY,80
 dyff/schema/adapters.py,sha256=YMTHv_2VlLGFp-Kqwa6H51hjffHmk8gXjZilHysIF5Q,123
 dyff/schema/annotations.py,sha256=nE6Jk1PLqlShj8uqjE_EzZC9zYnTDW5AVtQcjysiK8M,10018
 dyff/schema/base.py,sha256=jvaNtsSZyFfsdUZTcY_U-yfLY5_GyrMxSXhON2R9XR0,119
@@ -29,7 +29,7 @@ dyff/schema/v0/r1/adapters.py,sha256=hpwCSW8lkMkUKCLe0zaMUDu-VS_caSxJvPsECEi_XRA
 dyff/schema/v0/r1/base.py,sha256=1VJXVLKldOq3aH2HdVgxXXk4DJTZEIiaTa4GzMKzziU,20228
 dyff/schema/v0/r1/commands.py,sha256=jqbEvDRLpcVGWXqlbniuSKg3UIjlrpKt6_0hiwUpPfQ,20153
 dyff/schema/v0/r1/oci.py,sha256=YjHDVBJ2IIxqijll70OK6pM-qT6pq8tvU7D3YB9vGM0,6700
-dyff/schema/v0/r1/platform.py,sha256=Slh548oqJhfTfriz62YRut3WriGgE4FfwflG5oMwkTM,110503
+dyff/schema/v0/r1/platform.py,sha256=bQO3gQ6G-7Cl_FHHNLW1ozAG-ShYPOTIf-NfB6uvnps,110880
 dyff/schema/v0/r1/requests.py,sha256=JMpdx9WBlUpePshacG85XMmlGPBkkaK5453VaK3kH9s,18321
 dyff/schema/v0/r1/responses.py,sha256=nxy7FPtfw2B_bljz5UGGuSE79HTkDQxKH56AJVmd4Qo,1287
 dyff/schema/v0/r1/test.py,sha256=X6dUyVd5svcPCI-PBMOAqEfK9jv3bRDvkQTJzwS96c0,10720
@@ -43,9 +43,9 @@ dyff/schema/v0/r1/dataset/text.py,sha256=MYG5seGODDryRSCy-g0Unh5dD0HCytmZ3FeElC-
 dyff/schema/v0/r1/dataset/vision.py,sha256=tJFF4dkhHX0UXTj1sPW-G22xTSI40gbYO465FuvmvAU,443
 dyff/schema/v0/r1/io/__init__.py,sha256=L5y8UhRnojerPYHumsxQJRcHCNz8Hj9NM8b47mewMNs,92
 dyff/schema/v0/r1/io/vllm.py,sha256=vWyLg-susbg0JDfv6VExBpgFdU2GHP2a14ChOdbckvs,5321
-dyff_schema-0.38.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-dyff_schema-0.38.1.dist-info/licenses/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
-dyff_schema-0.38.1.dist-info/METADATA,sha256=LwpS_MihYVnxHSp9cuKFpjY72GTrHqgHuR5b7ztQM9Y,3734
-dyff_schema-0.38.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dyff_schema-0.38.1.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
-dyff_schema-0.38.1.dist-info/RECORD,,
+dyff_schema-0.38.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+dyff_schema-0.38.3.dist-info/licenses/NOTICE,sha256=YONACu0s_Ui6jNi-wtEsVQbTU1JIkh8wvLH6d1-Ni_w,43
+dyff_schema-0.38.3.dist-info/METADATA,sha256=LRDYIEaUA4AMUGCTGCLb1ABwrCh6rT28AU8ywiYU60o,3734
+dyff_schema-0.38.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dyff_schema-0.38.3.dist-info/top_level.txt,sha256=9e3VVdeX73t_sUJOPQPCcGtYO1JhoErhHIi3WoWGcFI,5
+dyff_schema-0.38.3.dist-info/RECORD,,