-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathsetup.py
36 lines (31 loc) · 968 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
28
29
30
31
32
33
34
35
from setuptools import setup
import re
def get_property(prop, project):
"""
Credits: https://stackoverflow.com/a/41110107
"""
result = re.search(r'{}\s*=\s*[\'"]([^\'"]*)[\'"]'.format(prop), open(project + '/__init__.py').read())
return result.group(1)
with open("README.md", "r") as f:
long_description = f.read()
setup(
name="textgenie",
version=get_property('__version__', "textgenie"),
description="A python library to augment text data using NLP.",
long_description=long_description,
long_description_content_type="text/markdown",
author="Het Pandya",
url="http://github.com/hetpandya/TextGenie",
license="MIT",
install_requires=[
"torch>=1.5.0",
"transformers",
"sentencepiece",
"spacy",
"tqdm",
"pandas",
"Pattern @ git+https://github.com/clips/pattern.git",
],
packages=["textgenie"],
)