nits

huggingface · Jan 19, 2024 · 20c4e82 · 20c4e82
1 parent 8f9b945
commit 20c4e82
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/bindings/python/Makefile b/bindings/python/Makefile
@@ -8,12 +8,14 @@ check_dirs := examples py_src/tokenizers tests
 # Format source code automatically
 style:
 	python stub.py
-	black --line-length 119 --target-version py35 $(check_dirs)
+	ruff check  $(check_dirs) --fix 
+	ruff format --line-length 119 --target-version py35 $(check_dirs)
 
 # Check the source code is formatted correctly
 check-style:
 	python stub.py --check
-	black --check --line-length 119 --target-version py35 examples py_src/tokenizers tests
+	ruff check examples py_src/tokenizers tests 
+	ruff format --check --line-length 119 --target-version py35 examples py_src/tokenizers tests 
 
 TESTS_RESOURCES = $(DATA_DIR)/small.txt $(DATA_DIR)/roberta.json
 

diff --git a/bindings/python/stub.py b/bindings/python/stub.py
@@ -121,6 +121,15 @@ def py_file(module, origin):
         string += f"{name} = {origin}.{name}\n"
     return string
 
+import subprocess
+from typing import List, Optional, Tuple
+
+def do_ruff(code):
+    command = ["ruff", "format", "-", "--config", "pyproject.toml", "--silent"]
+    process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE)
+    stdout, _ = process.communicate(input=code.encode())
+    return stdout.decode()
+
 
 def do_black(content, is_pyi):
     mode = black.Mode(