diff --git a/setup.py b/setup.py
index 1e89eb00..9649dcb1 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,8 @@
         'sympy',
         'pyyaml',
         'opt_einsum',
-        'scikit-learn'
+        'scikit-learn',
+        'rich'
     ],
     classifiers=[
         'Development Status :: 5 - Production/Stable',