contracode/setup.py at master · parasj/contracode · GitHub

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
41
from setuptools import setup

setup(
    name='representjs',
    version='1.0',
    packages=["representjs"],
    python_requires=">=3.7",
    install_requires=[
        "fire",
        "graphviz",
        "jsbeautifier",
        "jsonlines",
        "pyjsparser",
        "tqdm",
        "requests",
        "regex",
        "loguru",
        "pyarrow",

        # Data
        "matplotlib",
        "numpy",
        "pandas",
        "seaborn",

        # PyTorch
        "pytorch-lightning",
        "torch",
        "torchtext",
        "wandb",

        # NLP dependencies
        "sentencepiece",
        "sacremoses",
        "transformers>=3.2.0",
        "tokenizers",
        "datasets",
        "allennlp",
    ],
    extras_require={"test": ["pytest"]}
)