diff --git a/setup.py b/setup.py index 6135616235..e02f1c9641 100644 --- a/setup.py +++ b/setup.py @@ -502,7 +502,8 @@ def get_extensions(): # Remove csrc/cpu/*.cpp excluded_sources = list( glob.glob( - os.path.join(extensions_dir, "cpu/aten_kernels/*.cpp"), recursive=False + os.path.join(extensions_dir, "cpu", "aten_kernels", "*.cpp"), + recursive=False, ) ) sources = [s for s in sources if s not in excluded_sources]