Skip to content

Commit 8d20e10

Browse files
Deferred schema renderer creation to avoid requiring pyyaml. (#6416)
Closes #6366.
1 parent f539c0d commit 8d20e10

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

rest_framework/management/commands/generateschema.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,10 @@ def handle(self, *args, **options):
3232
self.stdout.write(output.decode('utf-8'))
3333

3434
def get_renderer(self, format):
35-
return {
36-
'corejson': CoreJSONRenderer(),
37-
'openapi': OpenAPIRenderer(),
38-
'openapi-json': JSONOpenAPIRenderer()
35+
renderer_cls = {
36+
'corejson': CoreJSONRenderer,
37+
'openapi': OpenAPIRenderer,
38+
'openapi-json': JSONOpenAPIRenderer,
3939
}[format]
40+
41+
return renderer_cls()

0 commit comments

Comments
 (0)