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'#