Let ast.parse deal with encoding issues.

Fixes #4
coherent-oss · Aug 31, 2024 · cd841c6 · cd841c6
1 parent b8fd744
commit cd841c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/imports.py b/imports.py
@@ -76,7 +76,7 @@ def _check_standard(top_level_name):
 
 
 @functools.singledispatch
-def get_module_imports(module: pathlib.Path | str) -> Generator[str]:
+def get_module_imports(module: pathlib.Path | str | bytes) -> Generator[str]:
     r"""
     Parse a Python module to extract imported names.
 
@@ -105,7 +105,7 @@ def get_module_imports(module: pathlib.Path | str) -> Generator[str]:
 
 @get_module_imports.register
 def _(module: pathlib.Path):
-    return get_module_imports(module.read_text())
+    return get_module_imports(module.read_bytes())
 
 
 def print_module_imports(path: pathlib.Path):