omni.isaac.lab.utils.io.yaml 源代码

# Copyright (c) 2022-2024, The Isaac Lab Project Developers.
# All rights reserved.
#
# SPDX-License-Identifier: BSD-3-Clause

"""Utilities for file I/O with yaml."""

import os
import yaml

from omni.isaac.lab.utils import class_to_dict


[文档]def load_yaml(filename: str) -> dict: """Loads an input PKL file safely. Args: filename: The path to pickled file. Raises: FileNotFoundError: When the specified file does not exist. Returns: The data read from the input file. """ if not os.path.exists(filename): raise FileNotFoundError(f"File not found: {filename}") with open(filename) as f: data = yaml.full_load(f) return data
[文档]def dump_yaml(filename: str, data: dict | object, sort_keys: bool = False): """Saves data into a YAML file safely. Note: The function creates any missing directory along the file's path. Args: filename: The path to save the file at. data: The data to save either a dictionary or class object. sort_keys: Whether to sort the keys in the output file. Defaults to False. """ # check ending if not filename.endswith("yaml"): filename += ".yaml" # create directory if not os.path.exists(os.path.dirname(filename)): os.makedirs(os.path.dirname(filename), exist_ok=True) # convert data into dictionary if not isinstance(data, dict): data = class_to_dict(data) # save data with open(filename, "w") as f: yaml.dump(data, f, default_flow_style=False, sort_keys=sort_keys)