fixed caching

c483a663 · rusty1s · 70afd38e · c483a663
Commit c483a663 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_cluster/src/cpu.h']
 sources = ['torch_cluster/src/cpu.c']
 include_dirs = ['torch_cluster/src']