fix docstring check

tserg · May 3, 2024 · 04db590 · 04db590
1 parent 2e6fbe7
commit 04db590
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/vyper/ast/parse.py b/vyper/ast/parse.py
@@ -235,7 +235,11 @@ def _visit_docstring(self, node):
 
         if node.body:
             n = node.body[0]
-            if isinstance(n, python_ast.Expr) and isinstance(n.value, python_ast.Str):
+            if (
+                isinstance(n, python_ast.Expr)
+                and isinstance(n.value, python_ast.Constant)
+                and isinstance(n.value.value, str)
+            ):
                 self.generic_visit(n.value)
                 n.value.ast_type = "DocStr"
                 del node.body[0]