Skip to content

Commit 728877a

Browse files
committed
fix: Don't wrap if print_invalid_expressions_as_is flag is set
1 parent a7b916e commit 728877a

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

pybind11_stubgen/__init__.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,6 +249,11 @@ def stub_parser_from_args(args: CLIArgs) -> IParser:
249249
),
250250
]
251251

252+
if args.print_invalid_expressions_as_is:
253+
wrap_invalid_expressions = []
254+
else:
255+
wrap_invalid_expressions = [WrapInvalidExpressionInAnnotated]
256+
252257
class Parser(
253258
*error_handlers_top, # type: ignore[misc]
254259
FixMissing__future__AnnotationsImport,
@@ -263,7 +268,7 @@ class Parser(
263268
FixMissingEnumMembersAnnotation,
264269
OverridePrintSafeValues,
265270
*numpy_fixes, # type: ignore[misc]
266-
WrapInvalidExpressionInAnnotated,
271+
*wrap_invalid_expressions,
267272
FixNumpyDtype,
268273
FixNumpyArrayFlags,
269274
FixCurrentModulePrefixInTypeNames,

0 commit comments

Comments
 (0)