dyff-schema 0.21.0__tar.gz → 0.22.0__tar.gz

{dyff_schema-0.21.0/dyff_schema.egg-info → dyff_schema-0.22.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: dyff-schema
-Version: 0.21.0
+Version: 0.22.0
 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.21.0 → dyff_schema-0.22.0}/dyff/schema/v0/r1/platform.py

@@ -1064,6 +1064,7 @@ class ModelStorageMedium(str, enum.Enum):
     Mock = "Mock"
     ObjectStorage = "ObjectStorage"
     PersistentVolume = "PersistentVolume"
+    FUSEVolume = "FUSEVolume"
 
 
 class ModelArtifactKind(str, enum.Enum):
@@ -1216,11 +1217,16 @@ class InferenceServiceRunner(DyffSchemaBaseModel):
     )
 
     accelerator: Optional[Accelerator] = pydantic.Field(
-        default=None, description="Optional accelerator hardware to use"
+        default=None, description="Optional accelerator hardware to use, per node."
     )
 
     resources: ModelResources = pydantic.Field(
-        description="Resource requirements to run the service."
+        description="Resource requirements to run the service, per node."
+    )
+
+    nodes: int = pydantic.Field(
+        default=1,
+        description="Number of nodes. The resource specs apply to *each node*.",
     )
 
 
@@ -1301,11 +1307,19 @@ class InferenceSessionBase(DyffSchemaBaseModel):
     replicas: int = pydantic.Field(default=1, description="Number of model replicas")
 
     accelerator: Optional[Accelerator] = pydantic.Field(
-        default=None, description="Accelerator hardware to use."
+        default=None, description="Accelerator hardware to use, per node."
     )
 
     useSpotPods: bool = pydantic.Field(
-        default=True, description="Use 'spot pods' for cheaper computation"
+        default=True,
+        description="Use preemptible 'spot pods' for cheaper computation."
+        " Note that some accelerator types may not be available in non-spot pods.",
+    )
+
+    nodes: int = pydantic.Field(
+        default=1,
+        ge=1,
+        description="Number of nodes. The resource specs apply to *each node*.",
     )
 
 

{dyff_schema-0.21.0 → dyff_schema-0.22.0/dyff_schema.egg-info}/PKG-INFO

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