Make schema_extra signature consistent

Author: r4victor

src/dstack/_internal/core/models/fleets.py

@@ -2,7 +2,7 @@
 import uuid
 from datetime import datetime
 from enum import Enum
-from typing import Any, Dict, List, Optional, Type, Union
+from typing import Any, Dict, List, Optional, Union
 
 from pydantic import Field, root_validator, validator
 from typing_extensions import Annotated, Literal
@@ -364,7 +364,7 @@ class FleetConfiguration(
 
 class FleetSpecConfig(CoreConfig):
     @staticmethod
-    def schema_extra(schema: Dict[str, Any], model: Type) -> None:
+    def schema_extra(schema: Dict[str, Any]):
         prop = schema.get("properties", {})
         prop.pop("merged_profile", None)
 

src/dstack/_internal/core/models/profiles.py

@@ -228,7 +228,7 @@ def crons(self) -> List[str]:
 
 class ProfileParamsConfig(CoreConfig):
     @staticmethod
-    def schema_extra(schema: Dict[str, Any]) -> None:
+    def schema_extra(schema: Dict[str, Any]):
         del schema["properties"]["pool_name"]
         del schema["properties"]["instance_name"]
         del schema["properties"]["retry_policy"]
@@ -413,7 +413,7 @@ class ProfileProps(CoreModel):
 
 class ProfileConfig(ProfileParamsConfig):
     @staticmethod
-    def schema_extra(schema: Dict[str, Any]) -> None:
+    def schema_extra(schema: Dict[str, Any]):
         ProfileParamsConfig.schema_extra(schema)
 
 

src/dstack/_internal/core/models/repos/remote.py

@@ -26,7 +26,7 @@ class RepoError(DstackError):
 
 class RemoteRepoCredsConfig(CoreConfig):
     @staticmethod
-    def schema_extra(schema: Dict[str, Any]) -> None:
+    def schema_extra(schema: Dict[str, Any]):
         del schema["properties"]["protocol"]
 
 
@@ -41,7 +41,7 @@ class RemoteRepoCreds(generate_dual_core_model(RemoteRepoCredsConfig)):
 
 class RemoteRepoInfoConfig(CoreConfig):
     @staticmethod
-    def schema_extra(schema: Dict[str, Any]) -> None:
+    def schema_extra(schema: Dict[str, Any]):
         del schema["properties"]["repo_host_name"]
         del schema["properties"]["repo_port"]
         del schema["properties"]["repo_user_name"]

src/dstack/_internal/core/models/runs.py

@@ -1,6 +1,6 @@
 from datetime import datetime, timedelta
 from enum import Enum
-from typing import Any, Dict, List, Literal, Optional, Type
+from typing import Any, Dict, List, Literal, Optional
 from urllib.parse import urlparse
 
 from pydantic import UUID4, Field, root_validator
@@ -394,7 +394,7 @@ class Job(CoreModel):
 
 class RunSpecConfig(CoreConfig):
     @staticmethod
-    def schema_extra(schema: Dict[str, Any], model: Type) -> None:
+    def schema_extra(schema: Dict[str, Any]):
         prop = schema.get("properties", {})
         prop.pop("merged_profile", None)
 

src/dstack/_internal/server/schemas/gateways.py

@@ -9,7 +9,7 @@
 
 class CreateGatewayRequestConfig(CoreConfig):
     @staticmethod
-    def schema_extra(schema: Dict[str, Any]) -> None:
+    def schema_extra(schema: Dict[str, Any]):
         del schema["properties"]["name"]
         del schema["properties"]["backend_type"]
         del schema["properties"]["region"]