diff --git a/rest_framework/management/commands/generateschema.py b/rest_framework/management/commands/generateschema.py index 214b66775..591073ba0 100644 --- a/rest_framework/management/commands/generateschema.py +++ b/rest_framework/management/commands/generateschema.py @@ -32,8 +32,10 @@ class Command(BaseCommand): self.stdout.write(output.decode('utf-8')) def get_renderer(self, format): - return { - 'corejson': CoreJSONRenderer(), - 'openapi': OpenAPIRenderer(), - 'openapi-json': JSONOpenAPIRenderer() + renderer_cls = { + 'corejson': CoreJSONRenderer, + 'openapi': OpenAPIRenderer, + 'openapi-json': JSONOpenAPIRenderer, }[format] + + return renderer_cls()