Fix configuration schema definitions

dstackai · Feb 8, 2024 · ccfb07f · ccfb07f
1 parent 1d5964d
commit ccfb07f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/dstack/_internal/core/models/resources.py b/src/dstack/_internal/core/models/resources.py
@@ -200,7 +200,8 @@ class Config:
         def schema_extra(schema: Dict[str, Any]):
             schema.clear()
             # replace strict schema with a more permissive one
-            for field, value in ResourcesSpecSchema.schema().items():
+            ref_template = "#/definitions/ResourcesSpec/definitions/{model}"
+            for field, value in ResourcesSpecSchema.schema(ref_template=ref_template).items():
                 schema[field] = value
 
     cpu: Range[int] = DEFAULT_CPU_COUNT