Skip to content

Commit 8129274

Browse files
committed
Improve the setup config
1 parent e303955 commit 8129274

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

.github/workflows/ci.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ jobs:
2121
run: |
2222
python -m pip install --upgrade pip
2323
pip install .
24-
pip install .[DL]
24+
pip install .[modeling]
2525
2626
- name: Lint with flake8
2727
run: |

setup.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,11 @@
2828
"torchvision==0.5",
2929
],
3030
extras_require={
31-
"GCV": ['google-cloud-vision'],
32-
"Tesseract": ['pytesseract'],
33-
"DL": [
31+
"ocr": [
32+
'google-cloud-vision',
33+
'pytesseract'
34+
],
35+
"modeling": [
3436
"pycocotools",
3537
"detectron2 @ git+https://github.com/facebookresearch/detectron2.git@v0.1.3#egg=detectron2",
3638
]

0 commit comments

Comments
 (0)