dyff-schema 0.36.3__tar.gz → 0.36.5__tar.gz

{dyff_schema-0.36.3/dyff_schema.egg-info → dyff_schema-0.36.5}/PKG-INFO

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

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

{dyff_schema-0.36.3 → dyff_schema-0.36.5}/dyff/schema/v0/r1/commands.py

@@ -543,6 +543,9 @@ __all__ = [
     "EditChallengeContentAttributes",
     "EditChallengeContentData",
     "EditChallengeContentPatch",
+    "EditChallengeTaskRulesAttributes",
+    "EditChallengeTaskRulesData",
+    "EditChallengeTaskRules",
     "EditEntityDocumentation",
     "EditEntityDocumentationAttributes",
     "EditEntityDocumentationData",

{dyff_schema-0.36.3 → dyff_schema-0.36.5}/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.3 → dyff_schema-0.36.5/dyff_schema.egg-info}/PKG-INFO

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

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