nat.experimental.test_time_compute.models.stage_enums

Classes

PipelineTypeEnum	Enum to represent the type of pipeline used in Inference Time Scaling.
StageTypeEnum	Enum to represent the type of stage in a pipeline.

Module Contents

class PipelineTypeEnum

Bases: str, enum.Enum

Enum to represent the type of pipeline used in Inference Time Scaling.

Initialize self. See help(type(self)) for accurate signature.

PLANNING = 'planning'

TOOL_USE = 'tool_use'

AGENT_EXECUTION = 'agent_execution'

CUSTOM = 'custom'

class StageTypeEnum

Bases: str, enum.Enum

Enum to represent the type of stage in a pipeline.

Initialize self. See help(type(self)) for accurate signature.

SEARCH = 'search'

EDITING = 'editing'

SCORING = 'scoring'

SELECTION = 'selection'

CUSTOM = 'custom'