From 29e82a03c5acca30dd68f73e461f24a561af1121 Mon Sep 17 00:00:00 2001 From: jiapeng Date: Tue, 31 Dec 2024 09:46:51 +0000 Subject: [PATCH] update setup --- README.md | 5 ++++- setup.py | 5 +++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index ef17edb..19cfecd 100644 --- a/README.md +++ b/README.md @@ -7,8 +7,11 @@ conda install tbb-devel ``` 2. If using python wrapper (Todo: automate pip installation) - a. Using pip: + a. Using pip (use the lightweight source code w/o data/notebooks): ``` + wget "https://github.com/PreferredAI/pcatt/archive/refs/tags/v0.10.tar.gz" + unzip pcatt-0.10.zip -d pcatt + cd pcatt pip install -r requirements.txt pip install . ``` diff --git a/setup.py b/setup.py index 54d9997..0423683 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,3 @@ -import pip from sysconfig import get_path from setuptools import setup, Extension @@ -6,7 +5,7 @@ module1 = Extension(f'greedy_builder', extra_compile_args = ["-O3", "-std=c++20"], define_macros = [('MAJOR_VERSION', '0'), - ('MINOR_VERSION', '1')], + ('MINOR_VERSION', '10')], include_dirs = [f'{PATH_PREFIX}/include/', f'{PATH_PREFIX}/include/tbb', f'{PATH_PREFIX}/include/oneapi'], @@ -22,5 +21,7 @@ author_email="jiapeng.lim.2021@phdcs.smu.edu.sg", license = "MIT", setup_requires=['pybind11', 'tbb-devel'], + url = "https://github.com/PreferredAI/pcatt/", + download_url = "https://github.com/PreferredAI/pcatt/archive/refs/tags/v0.10.tar.gz", ext_modules = [module1] )