-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathsetup.py
25 lines (24 loc) · 914 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
import os
from setuptools import setup, find_packages, find_namespace_packages
setup(
name='doc_enc',
version=os.environ.get('version', '0'),
description='Encoding texts as dense vectors',
author='dvzubarev',
author_email='[email protected]',
license='MIT',
packages=find_namespace_packages(include=["hydra_plugins.*"])
+ find_packages(include=["doc_enc*"]),
# packages=find_packages(),
install_requires=['numpy'],
entry_points={
'console_scripts': [
'run_training = doc_enc.training.run_training:train_cli',
'run_preproc = doc_enc.training.run_training:preproc_cli',
'run_repack = doc_enc.training.run_training:repack_cli',
'run_eval = doc_enc.eval.run_eval:eval_cli',
'docenccli = doc_enc.cli:main',
'fine_tune_classif = doc_enc.finetune_classif:fine_tune_classif_cli',
],
},
)