Lint

Author: larryliu0820

setup.py

@@ -10,23 +10,24 @@
 import re
 import subprocess
 import sys
-import shutil
 from pathlib import Path
 
-from setuptools import find_packages, setup, Extension
+from setuptools import Extension, find_packages, setup
 from setuptools.command.build_ext import build_ext
 
 # Read the README file
 with open("README.md", "r") as f:
     long_description = f.read()
 
+
 class CMakeExtension(Extension):
     def __init__(self, name, sourcedir=""):
         Extension.__init__(self, name, sources=[])
         self.sourcedir = os.path.abspath(sourcedir)
 
+
 class CMakeBuild(build_ext):
-    def build_extension(self, ext):
+    def build_extension(self, ext):  # noqa C901
         extdir = os.path.abspath(os.path.dirname(self.get_ext_fullpath(ext.name)))
 
         # Ensure the extension goes into the pytorch_tokenizers package directory
@@ -123,6 +124,7 @@ def build_extension(self, ext):
             ["cmake", "--build", "."] + build_args, cwd=build_temp, check=True
         )
 
+
 setup(
     name="pytorch-tokenizers",
     version="0.1.0",

test/test_python_bindings.py

@@ -9,7 +9,6 @@
 Test script for PyTorch Tokenizers Python bindings
 """
 
-import os
 import sys
 import unittest