# extended list of package dependencies to reach full functionality

# TODO: this shall be removed as we mock them in tests
comet-ml>=3.1.12
mlflow>=1.0.0
test_tube>=0.7.5

matplotlib>=3.1.1
# no need to install with [pytorch] as pytorch is already installed and torchvision is required only for Horovod examples
horovod>=0.19.2, != 0.20.0  # v0.20.0 has problem with building the wheel/installation
omegaconf>=2.0.0
# scipy>=0.13.3
scikit-learn>=0.22.2
torchtext>=0.3.1, <0.7  # TODO: temporary fix fix for compatibility
onnx>=1.7.0
onnxruntime>=1.3.0