-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
27 lines (24 loc) · 719 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import os
from skbuild import setup
from setuptools import find_packages
from setuptools.command.install import install
def set_cmake_args():
args = os.environ.get('BUILD_LIBTORCH', None)
if args :
return ["-DBUILD_LIBTORCH=ON"]
else:
return ["-DBUILD_LIBTORCH=OFF"]
setup(
name='SYRIPY',
version='1.0',
description='A package for modelling the generation and propagation of '
'synchrotron radiation ',
author='Robbie Watt',
author_email='rwatt1@stanford.edu',
packages=find_packages(),
cmake_args=set_cmake_args(),
cmake_install_dir="SYRIPY/Tracking/cTrack",
install_requires=[
'numpy',
'matplotlib',
'torch'])