From db47c31014bec0911a9a748f0afc0c40b7451938 Mon Sep 17 00:00:00 2001
From: Kerem Yilmaz <kerem@skyvern.com>
Date: Thu, 14 Mar 2024 12:55:02 -0700
Subject: [PATCH] Choose poetry env python version earlier (#71)

---
 setup.sh | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/setup.sh b/setup.sh
index 698f99ca..49a62d86 100755
--- a/setup.sh
+++ b/setup.sh
@@ -57,9 +57,14 @@ remove_poetry_env() {
     fi
 }
 
+# Choose python version
+choose_python_version_or_fail() {
+  poetry env use python3.11 || { echo "Error: Python 3.11 is not installed."; exit 1; }
+}
+
+
 # Function to install dependencies
 install_dependencies() {
-    poetry env use python3.11
     poetry install
 }
 
@@ -128,6 +133,7 @@ create_organization() {
 # Main function
 main() {
     initialize_env_file
+    choose_python_version_or_fail
     remove_poetry_env
     install_dependencies
     setup_postgresql