requirements.txt

# Library dependencies for Python code.  You need to install these with
# `pip install -r requirements.txt` or
# `conda install --file requirements.txt`
# to ensure that you can use all Snorkel code.
# NOTE: all essential packages must be placed under a section named
# '#### ESSENTIAL ...' so that the script `./scripts/check_requirements.py`
# can find them.

#### ESSENTIAL LIBRARIES

# General scientific computing
numpy>=1.16.0,<2.0.0
scipy>=1.2.0,<2.0.0

# Data storage and function application
pandas>=0.25.0,<0.26.0
tqdm>=4.33.0,<5.0.0

# Internal models
scikit-learn>=0.20.2,<0.22.0
torch>=1.1.0,<1.2.0
munkres==1.1.2

# LF dependency learning
networkx>=2.2,<2.4

# Model introspection tools
tensorboardX>=1.6,<2.0


#### EXTRA/TEST LIBRARIES

# NLP applications
spacy>=2.1.0,<3.0.0

# Scale-up/out
dask[complete]>=2.1.0,<3.0.0

# Serialization
dill>=0.3.0,<0.4.0


#### DEV TOOLS

black>=19.3b0,<20.0
flake8>=3.7.0,<4.0.0
isort>=4.3.0,<5.0.0
mypy>=0.720,<0.730
pydocstyle>=4.0.0,<5.0.0
pytest>=5.0.0,<6.0.0
pytest-cov>=2.7.0,<3.0.0
pytest-doctestplus>=0.3.0,<0.4.0
tox>=3.13.0,<4.0.0