diff --git a/mlem/cli/utils.py b/mlem/cli/utils.py index 4447dead..8bfec1a2 100644 --- a/mlem/cli/utils.py +++ b/mlem/cli/utils.py @@ -517,6 +517,7 @@ def _option_from_field( else NOT_SET, help=field.help, show_default=not field.required, + is_flag=type_ is bool, ) option.name = path return option diff --git a/tests/cli/test_declare.py b/tests/cli/test_declare.py index 9aedfd13..691d951d 100644 --- a/tests/cli/test_declare.py +++ b/tests/cli/test_declare.py @@ -303,24 +303,9 @@ class BooleanField(_MockBuilder): ( pytest.param( BooleanField(field=True), - "--field 1", - id="bool_true_1", - ), - pytest.param( - BooleanField(field=False), - "--field 0", - id="bool_false_0", - ), - pytest.param( - BooleanField(field=True), - "--field True", + "--field", id="bool_true", ), - pytest.param( - BooleanField(field=False), - "--field False", - id="bool_false", - ), ) )