dyff-schema 0.38.1__tar.gz → 0.38.3__tar.gz

{dyff_schema-0.38.1/dyff_schema.egg-info → dyff_schema-0.38.3}/PKG-INFO

@@ -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.3/dyff/schema/_version.py

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

{dyff_schema-0.38.1 → dyff_schema-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 → dyff_schema-0.38.3/dyff_schema.egg-info}/PKG-INFO

@@ -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/dyff/schema/_version.py

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