trinity.common.constants module

Constants.

class trinity.common.constants.CaseInsensitiveEnumMeta(cls, bases, classdict, **kwds)

Bases: EnumMeta

class trinity.common.constants.CaseInsensitiveEnum(value, *args, **kwargs)

Bases: Enum

An enumeration.

class trinity.common.constants.PromptType(value, *args, **kwargs)

Bases: CaseInsensitiveEnum

Prompt Type.

MESSAGES = 'messages'

CHATPAIR = 'chatpair'

PLAINTEXT = 'plaintext'

class trinity.common.constants.TaskType(value)

Bases: Enum

Task Type.

EXPLORE = 0

EVAL = 1

class trinity.common.constants.ReadStrategy(value, *args, **kwargs)

Bases: CaseInsensitiveEnum

Pop Strategy.

DEFAULT = None

FIFO = 'fifo'

RANDOM = 'random'

LRU = 'lru'

LFU = 'lfu'

PRIORITY = 'priority'

class trinity.common.constants.StorageType(value, *args, **kwargs)

Bases: CaseInsensitiveEnum

Storage Type.

SQL = 'sql'

QUEUE = 'queue'

FILE = 'file'

class trinity.common.constants.MonitorType(value, *args, **kwargs)

Bases: CaseInsensitiveEnum

Monitor Type.

WANDB = 'wandb'

TENSORBOARD = 'tensorboard'

class trinity.common.constants.SyncMethodEnumMeta(cls, bases, classdict, **kwds)

Bases: CaseInsensitiveEnumMeta

class trinity.common.constants.SyncMethod(value, *args, **kwargs)

Bases: CaseInsensitiveEnum

Sync Method.

NCCL = 'nccl'

CHECKPOINT = 'checkpoint'

MEMORY = 'memory'

class trinity.common.constants.RunningStatus(value)

Bases: Enum

Running status of explorer and trainer.

RUNNING = 'running'

REQUIRE_SYNC = 'require_sync'

WAITING_SYNC = 'waiting_sync'

STOPPED = 'stopped'

class trinity.common.constants.OpType(value)

Bases: Enum

Operator type for reward shaping.

ADD = 'add'

SUB = 'sub'

MUL = 'mul'

DIV = 'div'

class trinity.common.constants.SyncStyle(value, *args, **kwargs)

Bases: CaseInsensitiveEnum

An enumeration.

FIXED = 'fixed'

DYNAMIC_BY_TRAINER = 'dynamic_by_trainer'

DYNAMIC_BY_EXPLORER = 'dynamic_by_explorer'