From 19c714fe93fc070181d1ecfa87b5c67e156e3cc5 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Thu, 15 Jun 2017 14:38:18 +0200 Subject: [PATCH] Cleaning up. --- build.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/build.py b/build.py index da18d39..1de39c6 100755 --- a/build.py +++ b/build.py @@ -21,10 +21,10 @@ # You should have received a copy of the GNU General Public License # along with selector. If not, see . -import os +from os import path from torch.utils.ffi import create_extension -abs_path = os.path.dirname(os.path.abspath(__file__)) +abs_path = path.dirname(path.abspath(__file__)) ffi = create_extension( 'svrt', @@ -37,5 +37,4 @@ ffi = create_extension( with_cuda = False ) -if __name__ == '__main__': - ffi.build() +ffi.build() -- 2.39.5