diff --git a/setup.py b/setup.py index e5cc366..f43ba43 100644 --- a/setup.py +++ b/setup.py @@ -44,13 +44,14 @@ def read_requirements(): 'natsort', 'paramiko', 'pandas', + 'psutil', 'pytest', 'PyYAML', 'torch', 'torchvision', 'torchaudio', 'ipywidgets', - 'decord', + 'eva-decord', ]] deps_advanced = [deps_all_dict[dep] for dep in [