trinity.utils
Submodules
trinity.utils.distributed module
For distributed training with multiple process groups.
trinity.utils.dlc_utils module
trinity.utils.eval_utils module
- trinity.utils.eval_utils.find_boxed_answer(string)[source]
Find answers from solutions where the answers are enclosed in LaTeX’s oxed tag
- trinity.utils.eval_utils.extract_solution(solution_str)[source]
Extract the equation from the solution string.
trinity.utils.log module
Logging configuration for vLLM.
- class trinity.utils.log.NewLineFormatter(fmt, datefmt=None)[source]
Bases:
Formatter
Adds logging prefix to newlines to align multi-line messages.
- __init__(fmt, datefmt=None)[source]
Initialize the formatter with specified format strings.
Initialize the formatter either with the specified format string, or a default as described above. Allow for specialized date formatting with the optional datefmt argument. If datefmt is omitted, you get an ISO8601-like (or RFC 3339-like) format.
Use a style parameter of ‘%’, ‘{’ or ‘$’ to specify that you want to use one of %-formatting,
str.format()
({}
) formatting orstring.Template
formatting in your format string.Changed in version 3.2: Added the
style
parameter.
- format(record)[source]
Format the specified record as text.
The record’s attribute dictionary is used as the operand to a string formatting operation which yields the returned string. Before formatting the dictionary, a couple of preparatory steps are carried out. The message attribute of the record is computed using LogRecord.getMessage(). If the formatting string uses the time (as determined by a call to usesTime(), formatTime() is called to format the event time. If there is exception information, it is formatted using formatException() and appended to the message.
- trinity.utils.log.get_logger(name: str, level: int = 10) Logger [source]
Get a logger with the given name and level.
- Parameters:
name (str) – The name of the logger.
level (int, optional) – The level of the logger. Defaults to logging.DEBUG.
- Returns:
The logger with the given name and level.
- Return type:
logging.Logger
trinity.utils.monitor module
Monitor
- class trinity.utils.monitor.Monitor(project: str, name: str, role: str, config: Config | None = None)[source]
Bases:
ABC
- class trinity.utils.monitor.TensorboardMonitor(project: str, name: str, role: str, config: Config | None = None)[source]
Bases:
Monitor
trinity.utils.registry module
- class trinity.utils.registry.Registry(name: str)[source]
Bases:
object
This class is used to register some modules to registry by a repo name.
- property name
Get name of current registry.
- Returns:
name of current registry.
- property modules
Get all modules in current registry.
- Returns:
a dict storing modules in current registry.
- get(module_key)[source]
Get module named module_key from in current registry. If not found, return None.
- Parameters:
module_key – specified module name
- Returns:
module named module_key
- register_module(module_name: str | None = None, module_cls: type | None = None, force=False)[source]
Register module class object to registry with the specified modulename.
- Parameters:
module_name – module name
module_cls – module class object
force – Whether to override an existing class with the same name. Default: False.
Example
>>> registry = Registry() >>> @registry.register_module() >>> class TextFormatter: >>> pass
>>> class TextFormatter2: >>> pass >>> registry.register_module( module_name='text_formatter2', module_cls=TextFormatter2)