Class Workload

src/sol_execbench/core/data/workload.py:102–143

Concrete workload configuration for benchmarking. Defines a specific instance of a computational workload with concrete values for all variable axes and specifications for all input data. This represents an executable configuration that can be benchmarked.

class Workload(BaseModelWithDocstrings):
    """Concrete workload configuration for benchmarking.

    Defines a specific instance of a computational workload with concrete
    values for all variable axes and specifications for all input data.
    This represents an executable configuration that can be benchmarked.
    """

    axes: dict[str, NonNegativeInt]
    """Dictionary mapping axis names to their concrete integer values. All values must be
    positive."""
    inputs: dict[str, InputSpec]
    """Dictionary mapping input names to their data specifications."""
    uuid: NonEmptyString
    """Unique identifier for this specific workload configuration."""
    tolerance: ToleranceSpec = Field(default=ToleranceSpec())
    """Tolerance specification for the workload."""

    @model_validator(mode="after")
    def _validate_inputs(self) -> Workload:
        custom_inputs = [
            name
            for name, input in self.inputs.items()
            if isinstance(input, CustomInput)
        ]
        non_custom_inputs = [
            name
            for name, input in self.inputs.items()
            if not isinstance(input, CustomInput)
        ]
        if len(custom_inputs) > 0 and len(non_custom_inputs) > 0:
            raise ValueError(
                f"A workload cannot have both custom and non-custom inputs. Custom: {custom_inputs}. Non-custom: {non_custom_inputs}"
            )
        return self

    def get_scalar_inputs(self) -> dict[str, int]:
        return {
            name: input.value
            for name, input in self.inputs.items()
            if isinstance(input, ScalarInput)
        }