Add MapReduceWorkload scaffold and generic workload execution

This commit is contained in:
Emil
2026-08-02 01:02:24 +03:00
parent 19fbb8e926
commit bc76f386e5
21 changed files with 2241 additions and 1176 deletions
+4
View File
@@ -19,6 +19,7 @@ from .artifacts import (
PortSpec,
Provenance,
)
from .batch import MapReduceWorkload
from .conformance import (
CancellationFlag,
LocalArtifactStore,
@@ -77,6 +78,7 @@ from .registry import (
WorkloadDefinition,
WorkloadDescription,
WorkloadRegistry,
workload_allowlist_from_json,
)
from .resources import (
AcceleratorDevice,
@@ -155,6 +157,7 @@ __all__ = [
"LocalTaskContext",
"LoopSpec",
"MANIFEST_SCHEMA_VERSION",
"MapReduceWorkload",
"NegotiatedWorkload",
"NetworkPolicy",
"NumericTolerance",
@@ -207,6 +210,7 @@ __all__ = [
"WorkloadLimits",
"WorkloadManifest",
"WorkloadRegistry",
"workload_allowlist_from_json",
"assert_manifest_round_trip",
"installed_distribution_digest",
"negotiate_manifest",