numpy>=1.19.5
pandas>=1.1.5
sqlalchemy>=2.0.0
sqlalchemy-utils>=0.36.6
lark>=1.0.0
pyyaml>=5.1
aenum>=2.2.0
diskcache>=5.4.0
retry>=0.9.2
pydantic<2
psutil
thefuzz

[dev]
pytest>=6.1.2
pytest-cov>=2.11.1
mock
coveralls>=3.0.1
moto[s3]>=4.1.1
pytest-benchmark
codespell
pylint
black>=23.1.0
isort>=5.10.1
flake8>=3.9.1
wheel>=0.37.1
semantic_version
PyGithub
twine
PyDriller
torch>=1.10.0
torchvision>=0.11.1
transformers
faiss-cpu
opencv-python-headless>=4.6.0.66
Pillow>=8.4.0
eva-decord>=0.6.1
ultralytics>=8.0.93
timm>=0.6.13
sentencepiece
langchain
pymupdf<1.23.0
pdfminer.six
sentence-transformers
protobuf
bs4
openai>=0.27.4
gpt4all
facenet-pytorch>=2.5.2
pytube
youtube-transcript-api
boto3
norfair>=2.2.0
kornia
ipython<8.13.0
ipywidgets>=7.7.2
matplotlib>=3.3.4
nbmake>=1.2.1
nest-asyncio>=1.5.6

[document]
transformers
langchain
faiss-cpu
pymupdf<1.23.0
pdfminer.six
sentence-transformers
protobuf
bs4
openai>=0.27.4
gpt4all
sentencepiece

[ludwig]
ludwig[distributed,hyperopt]

[notebook]
ipython<8.13.0
ipywidgets>=7.7.2
matplotlib>=3.3.4
nbmake>=1.2.1
nest-asyncio>=1.5.6

[postgres]
psycopg2

[qdrant]
qdrant_client

[ray]
ray<2.5.0,>=1.13.0

[udf]
facenet-pytorch>=2.5.2
pytube
youtube-transcript-api
boto3
norfair>=2.2.0
kornia

[vision]
torch>=1.10.0
torchvision>=0.11.1
transformers
faiss-cpu
opencv-python-headless>=4.6.0.66
Pillow>=8.4.0
eva-decord>=0.6.1
ultralytics>=8.0.93
timm>=0.6.13
sentencepiece
