Skip to content

Commit

Permalink
update setup
Browse files Browse the repository at this point in the history
  • Loading branch information
jararap committed Dec 31, 2024
1 parent c59d5da commit 29e82a0
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 3 deletions.
5 changes: 4 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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 .
```
Expand Down
5 changes: 3 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
import pip
from sysconfig import get_path
from setuptools import setup, Extension

PATH_PREFIX = get_path('data')
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'],
Expand All @@ -22,5 +21,7 @@
author_email="[email protected]",
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]
)

0 comments on commit 29e82a0

Please sign in to comment.