from typing import Literal

from screensuite.benchmarks.hubbasebenchmark import HubBaseBenchmarkConfig


class AndroidWorldConfig(HubBaseBenchmarkConfig):
    hf_repo: Literal["smolagents/android_world"] = "smolagents/android_world"
    """HF repo name"""

    split: Literal["test", "train"] = "train"
    """Split to use."""

    max_steps: int = 5
    """Maximum number of steps to take."""
