trinity.common.constants module

Constants.

class trinity.common.constants.CaseInsensitiveEnumMeta(cls, bases, classdict, *, boundary=None, _simple=False, **kwds)

Bases: EnumType

name_aliases = {}

class trinity.common.constants.CaseInsensitiveEnum(new_class_name, /, names, *, module=None, qualname=None, type=None, start=1, boundary=None)

Bases: Enum

class trinity.common.constants.PromptType(*values)

Bases: CaseInsensitiveEnum

Prompt Type.

MESSAGES = 'messages'

PLAINTEXT = 'plaintext'

class trinity.common.constants.StorageType(*values)

Bases: CaseInsensitiveEnum

Storage Type.

SQL = 'sql'

QUEUE = 'queue'

FILE = 'file'

class trinity.common.constants.SyncMethodEnumMeta(cls, bases, classdict, *, boundary=None, _simple=False, **kwds)

Bases: CaseInsensitiveEnumMeta

name_aliases = {'offline': 'checkpoint', 'online': 'nccl'}

class trinity.common.constants.SyncMethod(*values)

Bases: CaseInsensitiveEnum

Sync Method.

NCCL = 'nccl'

CHECKPOINT = 'checkpoint'

MEMORY = 'memory'

class trinity.common.constants.RunningStatus(*values)

Bases: Enum

Running status of explorer and trainer.

RUNNING = 'running'

REQUIRE_SYNC = 'require_sync'

STOPPED = 'stopped'

class trinity.common.constants.OpType(*values)

Bases: Enum

Operator type for reward shaping.

ADD = 'add'

SUB = 'sub'

MUL = 'mul'

DIV = 'div'

class trinity.common.constants.SyncStyleEnumMeta(cls, bases, classdict, *, boundary=None, _simple=False, **kwds)

Bases: CaseInsensitiveEnumMeta

name_aliases = {'dynamic_by_explorer': 'explorer_driven', 'dynamic_by_trainer': 'trainer_driven'}

class trinity.common.constants.SyncStyle(*values)

Bases: CaseInsensitiveEnum

FIXED = 'fixed'

TRAINER_DRIVEN = 'trainer_driven'

EXPLORER_DRIVEN = 'explorer_driven'

class trinity.common.constants.SaveStrategy(*values)

Bases: CaseInsensitiveEnum

SINGLE_THREAD = 'single_thread'

SINGLE_PROCESS = 'single_process'

SINGLE_NODE = 'single_node'

UNRESTRICTED = 'unrestricted'