API ReferenceFull Library ReferenceNemo GymNemo Gym

nemo_gym.gitlab_utils

View as Markdown

Module Contents

Classes

NameDescription
MLFlowConfig-

Functions

API

class nemo_gym.gitlab_utils.MLFlowConfig()

Bases: BaseModel

mlflow_tracking_token
str
mlflow_tracking_uri
str
nemo_gym.gitlab_utils.create_mlflow_client() -> mlflow.MlflowClient
 
nemo_gym.gitlab_utils.delete_model_from_gitlab(
    model_name: str
) -> 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.upload_jsonl_dataset(
    config: nemo_gym.config_types.UploadJsonlDatasetGitlabConfig
) -> None
 
nemo_gym.gitlab_utils.upload_jsonl_dataset_cli() -> None