Avoid crash when file in repo contains non-unicode characters

rapidsai · Dec 13, 2023 · 1d673aa · 1d673aa
1 parent 78ca9ec
commit 1d673aa
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/rapids_dependency_file_generator/rapids_dependency_file_generator.py b/src/rapids_dependency_file_generator/rapids_dependency_file_generator.py
@@ -35,9 +35,12 @@ def delete_existing_files(root="."):
         for fn in filter(
             lambda fn: fn.endswith(".txt") or fn.endswith(".yaml"), filenames
         ):
-            with open(file_path := os.path.join(dirpath, fn)) as f:
-                if HEADER in f.read():
-                    os.remove(file_path)
+            try:
+                with open(file_path := os.path.join(dirpath, fn)) as f:
+                    if HEADER in f.read():
+                        os.remove(file_path)
+            except UnicodeDecodeError:
+                pass
 
 
 def dedupe(dependencies):