diff --git a/setup.py b/setup.py index 44d10014..6e325a57 100644 --- a/setup.py +++ b/setup.py @@ -12,8 +12,9 @@ def get_dist(pkgname): 'numpy', 'regex', 'tqdm', - 'gym' - + 'gym', + 'matplotlib', + 'opencv-python' ] tf_ver = '2.0.0a' if get_dist('tensorflow>='+tf_ver) is None and get_dist('tensorflow_gpu>='+tf_ver) is None: