From 7c96969632f66f83a0a02dd6b1cb1b06b6564b09 Mon Sep 17 00:00:00 2001 From: RichieHakim Date: Mon, 29 Jan 2024 19:36:45 -0500 Subject: [PATCH] Refactor dependency handling in setup.py --- setup.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/setup.py b/setup.py index 5a2d766..98f6597 100644 --- a/setup.py +++ b/setup.py @@ -43,7 +43,7 @@ def read_requirements(): ## Make different versions of dependencies ### Also pull out the version number from the requirements (specified in deps_all_dict values). -deps_core = [deps_all_dict[dep] for dep in [ +deps_core = {dep: deps_all_dict[dep] for dep in [ 'numpy', 'scipy', 'kornia', @@ -66,9 +66,9 @@ def read_requirements(): 'torchaudio', 'ipywidgets', 'eva-decord', -]] +]} -deps_advanced = [deps_all_dict[dep] for dep in [ +deps_advanced = {dep: deps_all_dict[dep] for dep in [ 'tables', 'opt-einsum', # 'rolling-quantiles', @@ -86,8 +86,10 @@ def read_requirements(): 'jupyter', 'PyWavelets', 'mat73', -]] +]} +deps_core_latest = dict(zip(deps_core.keys(), deps_core.keys())) +deps_advanced_latest = dict(zip(deps_advanced.keys(), deps_advanced.keys())) print({ 'deps_all': deps_all, @@ -125,6 +127,6 @@ def read_requirements(): extras_require={ 'all': list(deps_all_dict.values()), 'all_latest': list(deps_all_latest.values()), - 'core': deps_core, + 'core': list(deps_core.values()), }, ) \ No newline at end of file