From 9fb8e06813899102cb59bd9bf7f061f17e47661e Mon Sep 17 00:00:00 2001
From: Yusuf Kaka <yusufkaka@microsoft.com>
Date: Fri, 3 Feb 2023 16:42:42 +0200
Subject: [PATCH] Added missing dependencies to setup.py

---
 setup.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b723f6b..b39ae5d 100644
--- a/setup.py
+++ b/setup.py
@@ -22,6 +22,14 @@
     ],
     package_dir={"": "src"},
     packages=setuptools.find_packages(where="src"),
-    python_requires=">=3.6",
+    install_requires=[
+          'pyyaml',
+          'regex',
+          'matplotlib',
+          'plotly',
+          'scipy',
+          'scikit-learn',
+      ],
+    python_requires=">=3.6, <3.12",
     package_data={'prompt_engine': ['utils/encoder.json', 'utils/vocab.bpe']},
 )
\ No newline at end of file