Don't include doc string of class in its code child

diff --git a/Lib/compiler/transformer.py b/Lib/compiler/transformer.py
index 7f6a1b4a2c7d22ba67fe500da7071f15ebd090ca..7f615958daba449ad51370ae26cf25f5f52468da 100644 (file)
--- a/Lib/compiler/transformer.py
+++ b/Lib/compiler/transformer.py
@@ -222,6 +222,11 @@ class Transformer:
         # code for class
         code = self.com_node(nodelist[-1])
 
+        if doc is not None:
+            assert isinstance(code, Stmt)
+            assert isinstance(code.nodes[0], Discard)
+            del code.nodes[0]
+
         n = Class(name, bases, doc, code)
         n.lineno = nodelist[1][2]
         return n

diff --git a/Tools/compiler/compiler/transformer.py b/Tools/compiler/compiler/transformer.py
index 7f6a1b4a2c7d22ba67fe500da7071f15ebd090ca..7f615958daba449ad51370ae26cf25f5f52468da 100644 (file)
--- a/Tools/compiler/compiler/transformer.py
+++ b/Tools/compiler/compiler/transformer.py
@@ -222,6 +222,11 @@ class Transformer:
         # code for class
         code = self.com_node(nodelist[-1])
 
+        if doc is not None:
+            assert isinstance(code, Stmt)
+            assert isinstance(code.nodes[0], Discard)
+            del code.nodes[0]
+
         n = Class(name, bases, doc, code)
         n.lineno = nodelist[1][2]
         return n