pip>=21.0

# DATA SCIENCE & ML
# =================
numpy==1.19.5
pandas==1.2.2

# opencv-python==4.5.1.48

torch==1.7.1
torchvision==0.8.2
torch-optimizer==0.1.0
pytorch-lightning==1.1.7
# pytorch-lightning-bolts==0.3.0

# tensorflow==2.4.1
# tensorflow-gpu==2.4.1
# tensorboard==2.4.1

scipy==1.6.0
scikit-learn==0.24.1
# xgboost==1.3.3
# lightgbm==3.1.1

# IMAGE AUGMENTATION
# ==================
kornia==0.4.1  # $ pip install git+https://github.com/kornia/kornia
# imgaug==0.4.0
# albumentations==0.5.2

# INPUT / OUTPUT
# ================
# psutil==5.8.0
# pillow==8.1.0
imageio==2.9.0  # required for wandb video logging
moviepy==1.0.3  # required for wandb video logging
h5py>=2.10.0  # as of tensorflow 2.4 it does not support h5py 3+

# GRAPHING & DISPLAY
# ==================
matplotlib==3.3.4
plotly==4.14.3
# seaborn==0.11.1
tqdm==4.57.0

# streamlit==0.75.0
wandb==0.10.19

# UTILITY
# =======
# pytest==6.2.1

hydra-core==1.0.6
hydra-colorlog==1.0.0
# hydra-joblib-launcher==1.1.1
hydra-submitit-launcher==1.1.0

# submitit==1.1.5
