requirements-experiment.txt

# TODO: these requirements need to be cleaned up!

# DATA SCIENCE & ML
# =================
# pandas>=1.2.5
# torch-optimizer>=0.1.0  # optional install
# kornia>=0.5.5  # requires numpy<=1.9, but this can be relaxed or overridden

# requires pytorch to be installed first (duplicated in requirements-test.txt)
# torchsort>=0.1.4

# INPUT / OUTPUT
# ================
# psutil>=5.8.0
imageio>=2.9.0  # required for wandb video logging
moviepy>=1.0.3  # required for wandb video logging

# GRAPHING & DISPLAY
# ==================
matplotlib>=3
# seaborn>=0.11.1
wandb>=0.10.32

# UTILITY
# =======
omegaconf>=2.1.0   # only 2.1.0 supports nested variable interpolation eg. ${group.${group.key}}
hydra-core==1.1.1  # needs omegaconf
hydra-colorlog==1.1.0
hydra-submitit-launcher==1.1.6


# MISSING DEPS - these are imported or referened (_target_) in /experiments, but not included here OR in requirements.txt
# =============
# kornia