diff --git a/nemo_gym/server_utils.py b/nemo_gym/server_utils.py index 58562ea58..e263db430 100644 --- a/nemo_gym/server_utils.py +++ b/nemo_gym/server_utils.py @@ -45,7 +45,7 @@ from fastapi import FastAPI, Request, Response from fastapi.exception_handlers import request_validation_exception_handler from fastapi.exceptions import RequestValidationError -from fastapi.responses import JSONResponse, PlainTextResponse +from fastapi.responses import JSONResponse from omegaconf import DictConfig, OmegaConf, open_dict from pydantic import BaseModel, ConfigDict from requests.exceptions import ConnectionError @@ -548,7 +548,7 @@ class HeadServer(BaseServer): def setup_webserver(self) -> FastAPI: app = FastAPI() - app.get("/global_config_dict_yaml", response_class=PlainTextResponse)(self.global_config_dict_yaml) + app.get("/global_config_dict_yaml")(self.global_config_dict_yaml) app.get("/server_instances")(self.get_server_instances) return app