fix type errors

zsol · Jul 16, 2023 · 6f473b5 · 6f473b5
1 parent e381609
commit 6f473b5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/py_wtf/indexer/annotation.py b/py_wtf/indexer/annotation.py
@@ -126,6 +126,8 @@ def handle_string_like(
         contents = node.evaluated_value
         if not contents:
             return dumb_annotation(node)
+        if isinstance(contents, bytes):
+            contents = contents.decode()
         try:
             return self(cst.parse_expression(contents))
         except Exception as e:

diff --git a/py_wtf/indexer/file.py b/py_wtf/indexer/file.py
@@ -142,7 +142,10 @@ def extract_docstring(node: cst.Module | cst.BaseSuite) -> list[Documentation]:
         docstring = cast(cst.SimpleString, match["docstring"])
         docstring_quotes = {'"""', "'''"}
         if any(docstring.value.startswith(quote) for quote in docstring_quotes):
-            return [convert_to_myst(docstring.evaluated_value)]
+            value = docstring.evaluated_value
+            if isinstance(value, bytes):
+                value = value.decode()
+            return [convert_to_myst(value)]
     return []