import sys, os sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) # from lib import timed # from lib import pytorch_knn_cuda