Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/NVlabs/sage
/ functions
Functions
7,169 in github.com/NVlabs/sage
⨍
Functions
7,169
◇
Types & classes
1,266
↳
Endpoints
1
↓ 3,272 callers
Method
append
Append the data to the circular buffer. Args: data: The data to append to the circular buffer. The first dimension should be the
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/buffers/circular_buffer.py:107
↓ 502 callers
Method
load
Load the configuration file into memory. This function reads the contents of the configuration file into memory. If the file does not
IsaacLab/docker/utils/state_file.py:139
↓ 401 callers
Method
keys
Returns the keys of the scene entities. Returns: The keys of the scene entities.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/scene/interactive_scene.py:505
↓ 321 callers
Method
reset
(self)
M2T2/m2t2/m2t2_agent.py:42
↓ 295 callers
Method
mean
(self)
robomimic/robomimic/models/distributions.py:70
↓ 251 callers
Method
close
Run before terminating to make sure all logs are flushed
robomimic/robomimic/utils/log_utils.py:162
↓ 220 callers
Method
copy
(self)
robomimic/robomimic/config/config.py:257
↓ 198 callers
Method
update
(self, dt: float, force_recompute: bool = False)
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/sensors/imu/imu.py:105
↓ 198 callers
Method
values
(self)
robomimic/robomimic/models/distributions.py:94
↓ 179 callers
Method
reset
(self, env_ids: torch.Tensor | None = None)
IsaacLab/source/extensions/omni.isaac.lab/test/managers/test_observation_manager.py:57
↓ 168 callers
Method
meshgrid
(self, h, w)
matfuse-sd/src/ldm/models/diffusion/ddpm.py:819
↓ 165 callers
Method
text
(self)
server/vlm.py:276
↓ 160 callers
Method
step
(self, action: torch.Tensor)
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/envs/utils/marl.py:97
↓ 146 callers
Method
dump
Dumps the config to a json. Args: filename (str): if not None, save to json file. Returns: json_str
robomimic/robomimic/config/config.py:306
↓ 137 callers
Function
get_layout_from_scene_json_path
Load a room layout from JSON data and set it as the current layout.
server/utils.py:2905
↓ 133 callers
Method
deepcopy
(self)
robomimic/robomimic/config/config.py:260
↓ 128 callers
Method
add_app_launcher_args
Utility function to configure AppLauncher arguments with an existing argument parser object. This function takes an ``argparse.ArgumentParser
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/app/app_launcher.py:145
↓ 119 callers
Function
dump_yaml
Saves data into a YAML file safely. Note: The function creates any missing directory along the file's path. Args: filename:
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/io/yaml.py:33
↓ 118 callers
Method
close
(self)
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/envs/utils/marl.py:132
↓ 117 callers
Method
add
Add a key-value pair to the dataset. The key can be nested by using the "/" character. For example: "obs/joint_pos". Args:
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/datasets/episode_data.py:87
↓ 109 callers
Method
write_joint_state_to_sim
Write joint positions and velocities to the simulation. Args: position: Joint positions. Shape is (len(env_ids), len(joint_ids)).
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/articulation/articulation.py:470
↓ 103 callers
Function
dict_to_floor_plan
Convert a dictionary (from JSON) back to a FloorPlan object. Args: layout_data: Dictionary containing the floor plan data
server/isaacsim/isaac.sim.mcp_extension/isaac_sim_mcp_extension/scene/utils.py:309
↓ 103 callers
Function
get_forward_side_from_support_point_and_yaw
Given the support point (x, y) and yaw, it will return the forward and side movements. This is the inverse of the support_point function
server/isaaclab/omron_franka_occupancy.py:247
↓ 100 callers
Function
get_textured_object_mesh
(layout, room, room_id, object_id)
server/tex_utils.py:1566
↓ 94 callers
Function
export_layout_to_mesh_dict_object_id
(layout: FloorPlan, object_id: str)
server/tex_utils.py:815
↓ 94 callers
Method
make
Creates the encoder networks and locks the encoder so that more modalities cannot be added.
robomimic/robomimic/models/obs_nets.py:202
↓ 91 callers
Function
occupancy_map
It will return a function that can calculate the x-y occupancy for the robot. x-y occupancy means the area that the robot takes up in the x-
server/isaaclab/omron_franka_occupancy.py:4
↓ 91 callers
Method
save
Save the configuration file to disk.
IsaacLab/docker/utils/state_file.py:148
↓ 85 callers
Function
parse_env_cfg
Parse configuration for an environment and override based on inputs. Args: task_name: The name of the environment. device: The de
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/utils/parse_cfg.py:126
↓ 84 callers
Method
is_stopped
Whether data collection is stopped or not. Returns: True if data collection has stopped.
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/utils/data_collector/robomimic_data_collector.py:116
↓ 82 callers
Method
step
Step in the environment with an action. Args: action (np.array): action to take Returns: observatio
robomimic/robomimic/envs/env_gym.py:68
↓ 75 callers
Function
run_tests
Wrapper for running tests via ``unittest``. Args: verbosity: Verbosity level for the test runner. **kwargs: Additional arguments
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/app/runners.py:11
↓ 74 callers
Method
update
(self, step: int, replay_sample: dict)
M2T2/m2t2/m2t2_agent.py:39
↓ 73 callers
Function
build_simulation_context
Context manager to build a simulation context with the provided settings. This function facilitates the creation of a simulation context and prov
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/sim/simulation_context.py:705
↓ 70 callers
Function
load_m2t2
(cfg_path=cfg_path_default)
server/m2t2_utils/infer.py:42
↓ 70 callers
Method
write
(self, message)
robomimic/robomimic/utils/log_utils.py:30
↓ 69 callers
Function
export_layout_to_mesh_dict_list_tree_search_with_object_id
(layout: FloorPlan, object_id: str)
server/tex_utils.py:741
↓ 69 callers
Function
generate_m2t2_data
(mesh_dict_list, target_object_name, base_pos)
server/m2t2_utils/data.py:25
↓ 69 callers
Function
infer_m2t2
(meta_data, vis_data, model, cfg, return_contacts=False)
server/m2t2_utils/infer.py:94
↓ 68 callers
Method
info
(self, message: str)
server/augment/pose_aug_v0.py:67
↓ 67 callers
Method
write_root_velocity_to_sim
Set the root center of mass velocity over selected environment indices into the simulation. The velocity comprises linear velocity (x, y, z)
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/rigid_object/rigid_object.py:268
↓ 65 callers
Method
open
Open an existing dataset file.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/datasets/hdf5_dataset_file_handler.py:27
↓ 65 callers
Method
parameters
(self, trainable_only=False)
matfuse-sd/src/ldm/modules/encoders/multicondition.py:70
↓ 64 callers
Method
clear
Clears the plot.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/ui/widgets/line_plot.py:132
↓ 63 callers
Function
dump_pickle
Saves data into a pickle file safely. Note: The function creates any missing directory along the file's path. Args: filename
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/io/pkl.py:32
↓ 60 callers
Method
compute
Compute the desired state of the robot's end-effector and the gripper.
IsaacLab/source/standalone/environments/state_machine/lift_cube_sm.py:222
↓ 59 callers
Method
write_root_link_state_to_sim
Set the root link state over selected environment indices into the simulation. The root state comprises of the cartesian position, quaternion
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/rigid_object/rigid_object.py:178
↓ 58 callers
Method
write_root_pose_to_sim
Set the root pose over selected environment indices into the simulation. The root pose comprises of the cartesian position and quaternion ori
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/rigid_object/rigid_object.py:192
↓ 57 callers
Function
call_vlm
Call VLM API with retry logic. Args: vlm_type: Type of VLM to use ("claude", "qwen", "openai", "glmv") model: Model name
server/vlm.py:432
↓ 53 callers
Function
extract_json_from_response
Robustly extract JSON content from Claude's response, handling various formats. Args: response_text: Raw response text from Clau
server/utils.py:2759
↓ 52 callers
Method
register_buffer
(self, name, attr)
matfuse-sd/src/ldm/models/diffusion/ddim.py:23
↓ 52 callers
Method
reset
(self, seed: int | None = None, options: dict[str, Any] | None = None)
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/envs/utils/marl.py:80
↓ 51 callers
Method
decode
Pass latents through decoder. Latents should be passed in to this function at train-time for backpropagation, but they can be
robomimic/robomimic/models/vae_nets.py:1148
↓ 50 callers
Method
add
Add a key-value pair to the dataset. The key can be nested by using the "/" character. For example: "obs/joint_pos". Currently only t
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/utils/data_collector/robomimic_data_collector.py:134
↓ 50 callers
Method
write_root_state_to_sim
Set the root state over selected environment indices into the simulation. The root state comprises of the cartesian position, quaternion orie
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/rigid_object/rigid_object.py:149
↓ 49 callers
Function
load_pickle
Loads an input PKL file safely. Args: filename: The path to pickled file. Raises: FileNotFoundError: When the specified file
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/io/pkl.py:13
↓ 49 callers
Method
set_world_poses_from_view
Set the poses of the camera from the eye position and look-at target position. Args: eyes: The positions of the camera's eye. Sha
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/sensors/camera/camera.py:330
↓ 48 callers
Method
set_joint_position_target
Set joint position targets into internal buffers. .. note:: This function does not apply the joint targets to the simulation. It
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/articulation/articulation.py:796
↓ 48 callers
Method
write_data_to_sim
Writes data to the simulator.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/asset_base.py:205
↓ 46 callers
Method
to_dict
(self)
robomimic/robomimic/config/config.py:244
↓ 45 callers
Method
reset
(self, soft: bool = False)
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/sim/simulation_context.py:431
↓ 45 callers
Method
update
Update this config using another config or nested dictionary.
robomimic/robomimic/config/config.py:270
↓ 44 callers
Method
run
Run the complete simulation sequence.
server/isaacsim/isaac.sim.mcp_extension/examples/g1.py:393
↓ 44 callers
Method
values_unlocked
A context scope for modifying a Config object. Within the scope, only values can be updated (new keys cannot be created). Upon
robomimic/robomimic/config/config.py:127
↓ 40 callers
Method
close
Stop recording and save the file at its current state.
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/utils/data_collector/robomimic_data_collector.py:239
↓ 40 callers
Method
copy
Copy artifacts from the running container to the host machine. Args: output_dir: The directory to copy the artifacts to. Defaults
IsaacLab/docker/utils/container_interface.py:169
↓ 40 callers
Method
plan_motion
( self, target_position, target_quat, # default_joints, )
server/isaaclab/curobo_tools/curobo_ik_planner.py:106
↓ 39 callers
Method
reset
Reset the internals of data logger.
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/utils/data_collector/robomimic_data_collector.py:124
↓ 39 callers
Method
set_camera_view
Set the location and target of the viewport camera in the stage. Note: This is a wrapper around the :math:`omni.isaac.core.utils.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/sim/simulation_context.py:323
↓ 39 callers
Method
step
(self)
IsaacLab/source/standalone/workflows/ray/tuner.py:82
↓ 38 callers
Method
flush
Flush the episode data based on environment indices. Args: env_ids: Environment indices to write data for. Defaults to (0).
IsaacLab/source/extensions/omni.isaac.lab_tasks/omni/isaac/lab_tasks/utils/data_collector/robomimic_data_collector.py:187
↓ 37 callers
Function
to_torch
Converts all numpy arrays and torch tensors in nested dictionary or list or tuple to torch tensors on device @device and returns a new neste
robomimic/robomimic/utils/tensor_utils.py:397
↓ 36 callers
Method
apply
Apply the noise to the data. Args: data: The data to apply the noise to. Shape is (num_envs, ...). Returns:
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/noise/noise_model.py:133
↓ 36 callers
Function
plan_robot_traj
plan the robot traj from start_pos, start_quat to end_pos, end_quat pos and quat are support_point_pos and support_point_quat plan the t
server/isaaclab/data_generation_mobile_manipulation_v4.py:2481
↓ 36 callers
Method
sleep
Attempt to sleep at the specified rate in hz.
IsaacLab/source/standalone/tools/record_demos.py:74
↓ 36 callers
Method
visualize
Update markers in the viewport. .. note:: If the prim `PointInstancer` is hidden in the stage, the function will simply return
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/markers/visualization_markers.py:217
↓ 35 callers
Function
export_layout_to_json
Export a FloorPlan object to JSON file. Args: layout: FloorPlan object to export export_path: Path where the JSON file w
server/utils.py:1525
↓ 33 callers
Function
get_layout_from_scene_save_dir
Load a room layout from JSON data and set it as the current layout.
server/utils.py:2885
↓ 32 callers
Method
get_command
Returns the command for the specified command term. Args: name: The name of the command term. Returns: The c
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/managers/command_manager.py:374
↓ 32 callers
Method
has_rtx_sensors
Returns whether the simulation has any RTX-rendering related sensors. This function returns the value of the simulation parameter ``"/isaacla
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/sim/simulation_context.py:278
↓ 31 callers
Method
render
Render from simulation to either an on-screen window or off-screen to RGB array. Args: mode (str): pass "human" for on-s
robomimic/robomimic/envs/env_gym.py:121
↓ 31 callers
Method
set_visibility
Sets the visibility of the markers. The method does this through the USD API. Args: visible: flag to set the visibility.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/markers/visualization_markers.py:195
↓ 30 callers
Function
visualize_multi_env_trajectory_planning
Visualize trajectory planning results for multiple environments in a single image. Args: room_bounds: tuple of (min_x, min_y, ma
server/isaaclab/data_generation_mobile_manipulation_v4.py:2002
↓ 28 callers
Method
compute
( self, control_action: ArticulationActions, joint_pos: torch.Tensor, joint_vel: torch.Tensor )
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/actuators/actuator_pd.py:211
↓ 28 callers
Method
find_bodies
Find bodies in the rigid body based on the name keys. Please check the :meth:`omni.isaac.lab.utils.string_utils.resolve_matching_names` funct
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/rigid_object/rigid_object.py:130
↓ 28 callers
Method
to_device
Move the algo to a different device.
robomimic/robomimic/algo/algo.py:136
↓ 27 callers
Method
stop
Stop timing.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/timer.py:123
↓ 26 callers
Method
clear_instance
(cls)
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/sim/simulation_context.py:544
↓ 26 callers
Method
device
Memory device for computation.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/asset_base.py:143
↓ 26 callers
Method
find_joints
Find joints in the articulation based on the name keys. Please see the :func:`omni.isaac.lab.utils.string.resolve_matching_names` function fo
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/assets/articulation/articulation.py:223
↓ 26 callers
Function
instantiate_from_config
(config)
matfuse-sd/src/ldm/util.py:112
↓ 25 callers
Function
generate_unique_id
Generate a unique ID with the given prefix.
server/utils.py:483
↓ 25 callers
Method
step
Steps the simulation. .. note:: This function blocks if the timeline is paused. It only returns when the timeline is playing.
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/sim/simulation_context.py:439
↓ 24 callers
Function
extract_wall_side_from_id
Extract wall direction from wall ID. Args: wall_id: Wall ID string (e.g., "wall_room_abc123_north") Returns:
server/utils.py:520
↓ 23 callers
Function
dict_to_floor_plan
Convert a dictionary (from JSON) back to a FloorPlan object. Args: layout_data: Dictionary containing the floor plan data
server/tex_utils.py:379
↓ 22 callers
Function
sample_uniform
Sample uniformly within a range. Args: lower: Lower bound of uniform range. upper: Upper bound of uniform range. size: Th
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/math.py:1313
↓ 21 callers
Method
record
Record data with logger. Args: k (str): key string v (float or image): value to store epoch: curr
robomimic/robomimic/utils/log_utils.py:99
↓ 21 callers
Function
recursive_dict_list_tuple_apply
Recursively apply functions to a nested dictionary or list or tuple, given a dictionary of {data_type: function_to_apply}. Args:
robomimic/robomimic/utils/tensor_utils.py:10
↓ 21 callers
Method
sample
Returns a batch of samples from the prior distribution. Args: n (int): this argument is used to specify the number
robomimic/robomimic/models/vae_nets.py:151
↓ 20 callers
Function
combine_frame_transforms
r"""Combine transformations between two reference frames into a stationary frame. It performs the following transformation operation: :math:`T_{0
IsaacLab/source/extensions/omni.isaac.lab/omni/isaac/lab/utils/math.py:750
next →
1–100 of 7,169, ranked by callers