File size: 1,900 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
"""NoOp builder implementation."""
from typing import Any, Dict, Optional
from wandb.sdk.launch.builder.abstract import AbstractBuilder
from wandb.sdk.launch.environment.abstract import AbstractEnvironment
from wandb.sdk.launch.errors import LaunchError
from wandb.sdk.launch.registry.abstract import AbstractRegistry
from .._project_spec import EntryPoint, LaunchProject
from ..agent.job_status_tracker import JobAndRunStatusTracker
class NoOpBuilder(AbstractBuilder):
"""NoOp builder."""
type = "noop"
def __init__(
self,
builder_config: Dict[str, Any],
environment: AbstractEnvironment,
registry: AbstractRegistry,
) -> None:
"""Initialize a NoOpBuilder."""
self.environment = environment
self.registry = registry
@classmethod
def from_config(
cls,
config: dict,
environment: AbstractEnvironment,
registry: AbstractRegistry,
verify: bool = True,
) -> "AbstractBuilder":
"""Create a noop builder from a config."""
return cls(config, environment, registry)
async def verify(self) -> None:
"""Verify the builder."""
raise LaunchError("Attempted to verify noop builder.")
async def build_image(
self,
launch_project: LaunchProject,
entrypoint: EntryPoint,
job_tracker: Optional[JobAndRunStatusTracker] = None,
) -> str:
"""Build the image.
For this we raise a launch error since it can't build.
"""
raise LaunchError(
"Attempted build with noop builder. Specify a builder in your launch config at ~/.config/wandb/launch-config.yaml.\n"
"Note: Jobs sourced from git repos and code artifacts require a builder, while jobs sourced from Docker images do not.\n"
"See https://docs.wandb.ai/guides/launch/create-job."
)
|