nemo_gym.hf_utils#

Module Contents#

Functions#

API#

nemo_gym.hf_utils.create_huggingface_client(token: str) huggingface_hub.HfApi#
nemo_gym.hf_utils.check_jsonl_format(file_path: str) bool#

Check for the presence of the expected keys in the dataset

nemo_gym.hf_utils.upload_jsonl_dataset(
config: nemo_gym.config_types.UploadJsonlDatasetHuggingFaceConfig,
) None#
nemo_gym.hf_utils.download_jsonl_dataset(
config: nemo_gym.config_types.DownloadJsonlDatasetHuggingFaceConfig,
) None#
nemo_gym.hf_utils.download_jsonl_dataset_cli() None#