nemo_gym.data.gitlab_utils

Module Contents

Name	Description
`create_mlflow_client`	-
`delete_model_from_gitlab`	-
`download_jsonl_dataset`	-
`gitlab_storage_extras`	The install extras GitLab dataset storage needs when `config` carries its credentials.
`has_gitlab_storage_credentials`	-
`is_model_in_gitlab`	-
`upload_jsonl_dataset`	-

nemo_gym.data.gitlab_utils.create_mlflow_client() -> mlflow.MlflowClient

nemo_gym.data.gitlab_utils.delete_model_from_gitlab(
    model_name: str
) -> None

nemo_gym.data.gitlab_utils.download_jsonl_dataset(
    config: nemo_gym.config.config_types.DownloadJsonlDatasetGitlabConfig
) -> None

nemo_gym.data.gitlab_utils.gitlab_storage_extras(
    config: typing.Any
) -> tuple[str, ...]

The install extras GitLab dataset storage needs when config carries its credentials.

A driver that installs Gym must add these, or a prepare step that downloads from GitLab fails.

nemo_gym.data.gitlab_utils.has_gitlab_storage_credentials(
    config: nemo_gym.config.config_types.MLFlowConfig
) -> bool

nemo_gym.data.gitlab_utils.is_model_in_gitlab(
    model_name: str
) -> bool

nemo_gym.data.gitlab_utils.upload_jsonl_dataset(
    config: nemo_gym.config.config_types.UploadJsonlDatasetGitlabConfig
) -> None

nemo_gym.data.gitlab_utils.__getattr__ = moved_attr_getter(__name__, {'upload_jsonl_dataset_cli': 'nemo_gym.cli.dataset',...