Tweak when we delete LOAD_CONST RETURN_VALUE

rocky · rocky · commit e4e3743de511 · 2024-11-28T08:03:32.000-05:00
diff --git a/uncompyle6/semantics/pysource.py b/uncompyle6/semantics/pysource.py
@@ -1212,6 +1212,7 @@ def build_ast(
         is_lambda=False,
         noneInNames=False,
         is_top_level_module=False,
+        compile_mode="exec",
     ) -> GenericASTTraversal:
         # FIXME: DRY with fragments.py
 
@@ -1262,7 +1263,6 @@ def build_ast(
                         load_const.kind == "LOAD_CONST"
                         and load_const.linestart is None
                         and load_const.attr is None
-                        or is_top_level_module
                     ):
                         # Delete LOAD_CONST (None) RETURN_VALUE
                         del tokens[-2:]
@@ -1371,6 +1371,7 @@ def code_deparse(
         co,
         is_lambda=is_lambda_mode(compile_mode),
         is_top_level_module=is_top_level_module,
+        compile_mode=compile_mode,
     )
 
     # XXX workaround for profiling
