diff --git a/src/drf_yasg/management/commands/generate_swagger.py b/src/drf_yasg/management/commands/generate_swagger.py index ba26e14..fc481ff 100644 --- a/src/drf_yasg/management/commands/generate_swagger.py +++ b/src/drf_yasg/management/commands/generate_swagger.py @@ -43,7 +43,7 @@ class Command(BaseCommand): ) parser.add_argument( '-f', '--format', dest='format', - default='', choices=('json', 'yaml'), + default='', choices=['json', 'yaml'], type=str, help='Output format. If not given, it is guessed from the output file extension and defaults to json.' )