diff --git a/src/mozilla_taskgraph/transforms/replicate.py b/src/mozilla_taskgraph/transforms/replicate.py index 4550b2e..cd52114 100644 --- a/src/mozilla_taskgraph/transforms/replicate.py +++ b/src/mozilla_taskgraph/transforms/replicate.py @@ -11,7 +11,7 @@ from taskcluster.exceptions import TaskclusterRestFailure from taskgraph.transforms.base import TransformSequence from taskgraph.util.attributes import attrmatch -from taskgraph.util.schema import Schema +from taskgraph.util.schema import LegacySchema from taskgraph.util.taskcluster import ( find_task_id, get_artifact, @@ -19,7 +19,7 @@ ) from voluptuous import ALLOW_EXTRA, Any, Optional, Required -REPLICATE_SCHEMA = Schema( +REPLICATE_SCHEMA = LegacySchema( { Required( "replicate", diff --git a/src/mozilla_taskgraph/transforms/scriptworker/release_artifacts.py b/src/mozilla_taskgraph/transforms/scriptworker/release_artifacts.py index c0066a1..6699083 100644 --- a/src/mozilla_taskgraph/transforms/scriptworker/release_artifacts.py +++ b/src/mozilla_taskgraph/transforms/scriptworker/release_artifacts.py @@ -10,13 +10,13 @@ from taskgraph.transforms.base import TransformSequence from taskgraph.transforms.task import task_description_schema -from taskgraph.util.schema import Schema +from taskgraph.util.schema import LegacySchema from taskgraph.util.workertypes import worker_type_implementation from voluptuous import Extra, Optional, Required transforms = TransformSequence() -release_artifacts_schema = Schema( +release_artifacts_schema = LegacySchema( { Required("worker-type"): task_description_schema["worker-type"], Optional("release-artifacts"): [str], diff --git a/src/mozilla_taskgraph/transforms/scriptworker/shipit/mark_as_shipped.py b/src/mozilla_taskgraph/transforms/scriptworker/shipit/mark_as_shipped.py index 75fabe4..8302a33 100644 --- a/src/mozilla_taskgraph/transforms/scriptworker/shipit/mark_as_shipped.py +++ b/src/mozilla_taskgraph/transforms/scriptworker/shipit/mark_as_shipped.py @@ -6,12 +6,12 @@ from textwrap import dedent from taskgraph.transforms.base import TransformSequence -from taskgraph.util.schema import Schema, optionally_keyed_by, resolve_keyed_by +from taskgraph.util.schema import LegacySchema, optionally_keyed_by, resolve_keyed_by from voluptuous import Extra, Optional transforms = TransformSequence() -mark_as_shipped_schema = Schema( +mark_as_shipped_schema = LegacySchema( { Optional("name"): str, Optional("shipit-product"): optionally_keyed_by("build-type", str),