Wrap query functions (get_rank, get_world_size, etc.) without memory check.

Load the content of a file from a path based on the file extension. Args: path: The path to the file to load. **kwargs: Additiona

Runs any assertion function on agent tools or user tools.

Compute sum of per-sample means across variable-length responses. Parameters: x: Flat tensor containing concatenated per-token values acr

Sync the user and assistant tools. Subclass should override this method if tools need to be synced.

Returns the current status of the SIM card.

Retrieves a plan directly by its unique ID. Args: plan_id: The unique identifier of the plan. Returns:

Get the user from the database. Args: user_id: The user id, such as 'sara_doe_496'. Returns: The user.

Get user from database.

Toggles Airplane Mode ON or OFF. If turning OFF, simulates a network search. Returns the new state of airplane_mode.

Toggles Data Saver mode. Returns the new state.

Toggles the Data Roaming setting. Returns the new state.

Execute weight update operations

All offsets start from the begining of the prompt.

Check if the user message is a stop message. Args: message: The user message. Returns: True if the user mess

Check if the reward is successful.

Check if the message is a valid user history message.

Fire release_memory_occupation on all engines (non-blocking). Returns a list of Ray ObjectRefs.

Get the type of a tool.

Seats already booked for this show (only confirmed bookings count).

Rough token estimate: count chars in content/tool results, divide by 3.

Forward non-communication calls without memory check.

Get a club from the database.

Get flight instance from database.

Get a funding source for an account.

Get the grant app permission actions for the mms issue task.

Retrieves a line directly by its unique ID. Args: line_id: The unique identifier of the line. Returns:

Get a player from the database.

Get the remove app permission actions for the mms issue task.

Check if the ticket is in 'placed' state (modifiable).

Current timestamp in YY-MM-DD-HH-MM.

Map MCore weight names to Hugging Face weight names. Args: mcore_weights_name: MCore weight name Returns:

Backward pass for DownProjFunction - Pure Python implementation. Forward: output = topk_weights * (input @ w2) (with topk_weight mul

Returns the full current state of the device attributes.

Find the longest cached prefix for the given text. Args: text: Input string to find prefix for Returns:

Returns the current date and time in the format YYYYMMDD_HHMMSS.

Returns options for the simulator.

Get the policy of the domain.

Get the response of the domain. This also calls sync_tools. Args: message: The message to get the response for. R

Return num_samples samples

Calculate correct sample mean for CP

Get all the tool signatures from a tool kit. Returns: A dictionary of tool signatures.

Get the tools of the domain.

Load a list of {question, label} dicts from a JSONL file; supports chat messages format.

Trigger post-process for int4/fp4 quantization on all rollout engines.

Print the trie structure in a readable format.

Reset internal offset to the start and return self.

Block forward pre-hook for certain configurations.

Return True when a periodic action (eval/save/checkpoint) should run. Args: rollout_id: The current rollout index (0-based).

Pause CUDA memory for all tracked tensors.

Validate the run config

Build BookedSeat list with computed price, convenience_fee, tax.

Same turns concatenation logic as agentflow.Solver: turn[0]: only response tokens are concatenated (prompt stored separately in pro

Checks if the default messaging app can send MMS messages.

Get an account from the database.

Get flight from database.

Get a saved payment instrument for a patron.