diff --git a/nvflare/fuel/utils/component_classes.json b/nvflare/fuel/utils/component_classes.json index 6c19777d1d..9981f61111 100644 --- a/nvflare/fuel/utils/component_classes.json +++ b/nvflare/fuel/utils/component_classes.json @@ -520,6 +520,9 @@ "TFModelPersistor": [ "nvflare.app_opt.tf.model_persistor" ], + "MLflowReceiver": [ + "nvflare.app_opt.tracking.mlflow.mlflow_receiver" + ], "MLflowWriter": [ "nvflare.app_opt.tracking.mlflow.mlflow_writer" ],