dyff-schema 0.38.7__tar.gz → 0.38.9__tar.gz

{dyff_schema-0.38.7/dyff_schema.egg-info → dyff_schema-0.38.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dyff-schema
-Version: 0.38.7
+Version: 0.38.9
 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.9/dyff/schema/_version.py

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

{dyff_schema-0.38.7 → dyff_schema-0.38.9}/dyff/schema/v0/r1/platform.py

@@ -2787,14 +2787,29 @@ class ChallengeSubmission(DyffEntity, ChallengeSubmissionBase):
 class PipelineEvaluationRequest(EvaluationRequestBase):
     kind: Literal["PipelineEvaluationRequest"] = "PipelineEvaluationRequest"
 
+    def dependencies(self) -> list[str]:
+        # This is a dirty hack that works because the types are structurally
+        # equivalent as far as .dependencies() cares
+        return Evaluation.dependencies(self)  # type: ignore[arg-type]
+
 
 class PipelineMeasurementRequest(AnalysisRequestBase):
     kind: Literal["PipelineMeasurementRequest"] = "PipelineMeasurementRequest"
 
+    def dependencies(self) -> list[str]:
+        # This is a dirty hack that works because the types are structurally
+        # equivalent as far as .dependencies() cares
+        return Measurement.dependencies(self)  # type: ignore[arg-type]
+
 
 class PipelineSafetyCaseRequest(AnalysisRequestBase):
     kind: Literal["PipelineSafetyCaseRequest"] = "PipelineSafetyCaseRequest"
 
+    def dependencies(self) -> list[str]:
+        # This is a dirty hack that works because the types are structurally
+        # equivalent as far as .dependencies() cares
+        return SafetyCase.dependencies(self)  # type: ignore[arg-type]
+
 
 PipelineNodeRequest = Union[
     PipelineEvaluationRequest,
@@ -2836,16 +2851,25 @@ class PipelineNode(DyffSchemaBaseModel):
 
 class PipelineParameter(DyffSchemaBaseModel):
     """Declares a parameter that can be passed to the pipeline to customize its
-    behavior."""
+    behavior.
+
+    By default, the argument will be substituted for the placeholder value
+    ``$(keyword)``, where ``keyword`` is the ``.keyword`` property of the
+    parameter. In this case, the value of the argument must be a string.
 
-    keyword: str = pydantic.Field(description="The keyword of the argument.")
-    destination: str = pydantic.Field(
+    If ``.destination`` is specified, the argument will instead overwrite
+    the value at the specified JSON path.
+    """
+
+    keyword: str = pydantic.Field(description="The keyword of the parameter.")
+    destination: Optional[str] = pydantic.Field(
+        default=None,
         description="The field in a pipeline node to substitute with the"
-        " parameter value. Should be a string like 'node_name.field1.field2'."
+        " parameter value. Should be a string like 'node_name.field1.field2'.",
     )
     description: Optional[str] = pydantic.Field(
         default=None,
-        description="A description of the argument.",
+        description="A description of the parameter.",
         max_length=summary_maxlen(),
     )
 
@@ -2856,6 +2880,10 @@ class PipelineBase(DyffSchemaBaseModel):
         description="The nodes in the pipeline graph.",
         min_length=1,
     )
+    parameters: dict[str, PipelineParameter] = pydantic.Field(
+        default_factory=dict,
+        description="Input parameters used to customize the pipeline.",
+    )
 
     @pydantic.field_validator("nodes", mode="after")
     def validate_node_names_match(cls, nodes: dict[str, PipelineNode]):
@@ -2864,6 +2892,15 @@ class PipelineBase(DyffSchemaBaseModel):
                 raise ValueError(f"nodes[{k}]: dict key must match value.name")
         return nodes
 
+    @pydantic.field_validator("parameters", mode="after")
+    def validate_parameter_keywords_match(
+        cls, parameters: dict[str, PipelineParameter]
+    ):
+        for k, v in parameters.items():
+            if k != v.keyword:
+                raise ValueError(f"parameters[{k}]: dict key must match value.keyword")
+        return parameters
+
 
 class Pipeline(DyffEntity, PipelineBase):
     """A set of Dyff workflows that can be executed as a group.
@@ -3166,6 +3203,7 @@ _DyffEntityTypeRevisable = Union[
     Audit,
     AuditProcedure,
     Challenge,
+    ChallengeSubmission,
     DataSource,
     Dataset,
     Evaluation,
@@ -3261,6 +3299,7 @@ _ENTITY_CLASS: dict[Entities, type[DyffEntityType]] = {
     Entities.Report: Report,
     Entities.Revision: Revision,
     Entities.SafetyCase: SafetyCase,
+    Entities.Submission: ChallengeSubmission,
     Entities.Team: Team,
     Entities.UseCase: UseCase,
 }
@@ -3276,6 +3315,7 @@ DyffEntityT = TypeVar(
     Audit,
     AuditProcedure,
     Challenge,
+    ChallengeSubmission,
     DataSource,
     Dataset,
     Evaluation,

{dyff_schema-0.38.7 → dyff_schema-0.38.9/dyff_schema.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dyff-schema
-Version: 0.38.7
+Version: 0.38.9
 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.7/dyff/schema/_version.py

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