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

jeremyhylton · jeremyhylton · commit da8db8ca186a · 2001-08-29T17:19:02.000Z
diff --git a/Lib/compiler/transformer.py b/Lib/compiler/transformer.py
@@ -222,6 +222,11 @@ def classdef(self, nodelist):
         # 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
@@ -222,6 +222,11 @@ def classdef(self, nodelist):
         # 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
