dyff-schema 0.36.2__tar.gz → 0.36.4__tar.gz

{dyff_schema-0.36.2/dyff_schema.egg-info → dyff_schema-0.36.4}/PKG-INFO

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

@@ -0,0 +1,2 @@
+__version__ = version = "0.36.4"
+__version_tuple__ = version_tuple = (0, 36, 4)

{dyff_schema-0.36.2 → dyff_schema-0.36.4}/dyff/schema/v0/r1/commands.py

@@ -147,13 +147,8 @@ class EditChallengeContentPatch(DyffSchemaBaseModel):
     )
 
 
-class EditChallengeContentAttributes(DyffSchemaBaseModel):
-    """Attributes for the EditChallengePageContent command."""
-
-    content: Optional[EditChallengeContentPatch] = pydantic.Field(
-        default=None,
-        description="Edits to make to the content of the main challenge page.",
-    )
+class EditChallengeContentAttributes(EditChallengeContentPatch):
+    """Attributes for the EditChallengeContent command."""
 
     tasks: Optional[dict[str, EditChallengeContentPatch]] = pydantic.Field(
         default=None, description="Edits to make to the content of the challenge tasks."
@@ -161,7 +156,7 @@ class EditChallengeContentAttributes(DyffSchemaBaseModel):
 
 
 class EditChallengeContentData(EntityIdentifier):
-    """Payload data for the EditChallengePageContent command."""
+    """Payload data for the EditChallengeContent command."""
 
     attributes: EditChallengeContentAttributes = pydantic.Field(
         description="The command attributes"

{dyff_schema-0.36.2 → dyff_schema-0.36.4}/dyff/schema/v0/r1/platform.py

@@ -25,7 +25,7 @@ import urllib.parse
 from datetime import datetime, timedelta, timezone
 from enum import Enum
 from pathlib import Path
-from typing import Any, Literal, NamedTuple, Optional, Type, Union
+from typing import Any, Literal, NamedTuple, Optional, Type, TypeVar, Union
 
 import i18naddress
 import pyarrow
@@ -227,6 +227,13 @@ class Entities(str, enum.Enum):
     Team = "Team"
     UseCase = "UseCase"
 
+    @staticmethod
+    def for_type(t: type["DyffEntityT"]) -> "Entities":
+        try:
+            return Entities(t.__name__)
+        except ValueError:
+            raise ValueError(f"not a DyffEntity type: {t}")
+
 
 class Resources(str, enum.Enum):
     """The resource names corresponding to entities that have API endpoints."""
@@ -2393,7 +2400,7 @@ class TeamMember(DyffSchemaBaseModel):
 
 
 class GoogleI18nMailingAddress(DyffSchemaBaseModel):
-    """An international mailing address, in the schema used by Google's
+    """An international mailing address, in the schema used by Google's main challenge page
     i18n address metadata repository
     (https://chromium-i18n.appspot.com/ssl-address) and as interpreted by the
     Python ``google-i18n-address`` package.
@@ -2945,6 +2952,8 @@ 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[
     Audit,
     AuditProcedure,
@@ -3016,9 +3025,37 @@ class Revision(DyffEntity, RevisionMetadata):
         return None
 
 
+# A Union type containing all Dyff top-level schema types
 DyffEntityType = Union[_DyffEntityTypeRevisable, History, Revision]
 
 
+# A TypeVar that matches any Dyff top-level schema type
+DyffEntityT = TypeVar(
+    "DyffEntityT",
+    Audit,
+    AuditProcedure,
+    Challenge,
+    DataSource,
+    Dataset,
+    Evaluation,
+    Family,
+    Hazard,
+    History,
+    InferenceService,
+    InferenceSession,
+    Measurement,
+    Method,
+    Model,
+    Module,
+    OCIArtifact,
+    Report,
+    Revision,
+    SafetyCase,
+    Team,
+    UseCase,
+)
+
+
 __all__ = [
     "SYSTEM_ATTRIBUTES",
     "Accelerator",
@@ -3064,6 +3101,7 @@ __all__ = [
     "Documented",
     "DyffDataSchema",
     "DyffEntity",
+    "DyffEntityT",
     "DyffEntityType",
     "DyffModelWithID",
     "DyffSchemaBaseModel",

{dyff_schema-0.36.2 → dyff_schema-0.36.4/dyff_schema.egg-info}/PKG-INFO

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

@@ -1,2 +0,0 @@
-__version__ = version = "0.36.2"
-__version_tuple__ = version_tuple = (0, 36, 2)