nemo_gym.artifacts.exporters.mlflow_config

The settings of gym eval export mlflow, kept apart from the exporter so that they can be imported without the MLflow SDK: gym eval submit validates an export’s settings on a machine that may not have it.

Module Contents

Classes

Name	Description
`MLflowExportConfig`	Settings of `gym eval export mlflow`.

API

class nemo_gym.artifacts.exporters.mlflow_config.MLflowExportConfig()

Bases: RunExportConfig, MLFlowConfig, UploadRolloutsConfigMixin

Settings of gym eval export mlflow.