fixed caching

0e98ec12 · rusty1s · 61eb4d03 · 0e98ec12
Commit 0e98ec12 authored Mar 03, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

build.py build.py +5 -0

No files found.
--- a/build.py
+++ b/build.py
+import os
+import shutil
 import subprocess

 import torch
 from torch.utils.ffi import create_extension

+if os.path.exists('build'):
+    shutil.rmtree('build')
+
 headers = ['torch_scatter/src/cpu.h']
 sources = ['torch_scatter/src/cpu.c']
 include_dirs = ['torch_scatter/src']