Try to update extend_parameters_schema to use a msgspec schema

Author: oskirby

taskcluster/mozillavpn_taskgraph/parameters.py

@@ -3,26 +3,24 @@
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
 import os
+from typing import Annotated, Optional
 
-from taskgraph.parameters import extend_parameters_schema
-from voluptuous import All, Any, Range, Required
+import msgspec
 
+from taskgraph.parameters import extend_parameters_schema
+from taskgraph.util.schema import Schema
 
 def get_defaults(repo_root):
     return {
         "pull_request_number": None,
         "version": "",
     }
 
+class MozillaVpnParametersSchema(Schema, kw_only=True, rename=None):
+    pull_request_number: Optional[Annotated[int, msgspec.Meta(ge=1)]]
+    version: str
 
-extend_parameters_schema(
-    {
-        Required("pull_request_number"): Any(All(int, Range(min=1)), None),
-        Required("version"): str,
-    },
-    defaults_fn=get_defaults,
-)
-
+extend_parameters_schema(MozillaVpnParametersSchema, defaults_fn=get_defaults)
 
 def get_decision_parameters(graph_config, parameters):
     head_tag = parameters["head_tag"]