-
Notifications
You must be signed in to change notification settings - Fork 7
/
setup.py
40 lines (31 loc) · 1.41 KB
/
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
28
29
30
31
32
33
34
35
36
37
38
39
40
import os
from setuptools import setup, find_packages
with open('README.md', 'r', encoding='utf-8') as f:
long_description = f.read()
def read(rel_path):
here = os.path.abspath(os.path.dirname(__file__))
# intentionally *not* adding an encoding option to open, See:
# https://github.com/pypa/virtualenv/issues/201#issuecomment-3145690
with open(os.path.join(here, rel_path), 'r') as fp:
return fp.read()
def get_version(rel_path):
for line in read(rel_path).splitlines():
if line.startswith('__version__'):
return line.split("'")[1]
raise RuntimeError('Unable to find version string.')
with open('requirements.txt', 'r') as requirements:
setup(name='drattack',
version=get_version('drattack/__init__.py'),
install_requires=list(requirements.read().splitlines()),
packages=find_packages(),
description='library for creating adversarial prompts for language models',
python_requires='>=3.6',
author='Xirui Li, Ruochen Wang, Minhao Cheng, J. Tianyi Zhou, Cho-Jui Hsieh',
author_email='xiruili@ucla.edu',
classifiers=[
'Programming Language :: Python :: 3',
'License :: OSI Approved :: MIT License',
'Operating System :: OS Independent'
],
long_description=long_description,
long_description_content_type='text/markdown')