`nemo_gym.gitlab_utils`#

Module Contents#

Classes#

Functions#

`create_mlflow_client`
`upload_jsonl_dataset`
`upload_jsonl_dataset_cli`
`download_jsonl_dataset`
`download_jsonl_dataset_cli`
`is_model_in_gitlab`
`delete_model_from_gitlab`

API#

class nemo_gym.gitlab_utils.MLFlowConfig#

Bases: pydantic.BaseModel

mlflow_tracking_uri: str#: None

mlflow_tracking_token: str#: None

nemo_gym.gitlab_utils.create_mlflow_client() → mlflow.MlflowClient#

nemo_gym.gitlab_utils.upload_jsonl_dataset( config: nemo_gym.config_types.UploadJsonlDatasetGitlabConfig, ) → None#

nemo_gym.gitlab_utils.upload_jsonl_dataset_cli() → None#

nemo_gym.gitlab_utils.download_jsonl_dataset( config: nemo_gym.config_types.DownloadJsonlDatasetGitlabConfig, ) → None#

nemo_gym.gitlab_utils.download_jsonl_dataset_cli() → None#

nemo_gym.gitlab_utils.is_model_in_gitlab(model_name: str) → bool#

nemo_gym.gitlab_utils.delete_model_from_gitlab(model_name: str) → None#