Class MockConfig

tests/worker/test_gpu_prompt_logprobs.py:41–65

# Mock classes and constants needed for the test
class MockConfig:

    class ModelConfig:
        enable_logprob = False
        max_logprobs = -1
        logprobs_mode = "raw_logprobs"
        enable_mm = False

    class SchedulerConfig:
        max_num_seqs = 6

    class ParallelConfig:
        enable_expert_parallel = False

    class CacheConfig:
        enable_prefix_caching = False

    class SpecaulativeConfig:
        method = None

    speculative_config = SpecaulativeConfig()
    model_config = ModelConfig()
    scheduler_config = SchedulerConfig()
    cache_config = CacheConfig()
    parallel_config = ParallelConfig()


class MockTask: