From 0afc175a4d7e6028d3bf7ad68836b46cefa40e17 Mon Sep 17 00:00:00 2001
From: tmadlener <thomas.madlener@desy.de>
Date: Wed, 31 Jul 2024 11:58:47 +0200
Subject: [PATCH] Fix the env.sh script to point to the correct python location

---
 env.sh | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/env.sh b/env.sh
index 014426fb0..1c0f250a7 100644
--- a/env.sh
+++ b/env.sh
@@ -25,8 +25,15 @@ elif [[ "$unamestr" = 'Darwin' ]]; then
     fi
 fi
 
-if ! echo $PYTHONPATH | grep -o $PODIO/python > /dev/null 2>&1; then
-  export PYTHONPATH=$PODIO/python:$PYTHONPATH
+python_stem=python3.$(python -c 'import sys; print(sys.version_info[1])')/site-packages
+if [ -d $PODIO/lib64/${python_stem} ]; then
+  if ! echo $PYTHONPATH | grep -o $PODIO/lib64/${python_stem} > /dev/null 2>&1; then
+    export PYTHONPATH=$PODIO/lib64/${python_stem}:$PYTHONPATH
+  fi
+else
+  if ! echo $PYTHONPATH | grep -o $PODIO/lib/${python_stem} > /dev/null 2>&1; then
+    export PYTHONPATH=$PODIO/lib/${python_stem}:$PYTHONPATH
+  fi
 fi
 
 if ! echo $ROOT_INCLUDE_PATH | grep -o $PODIO/include > /dev/null 2>&1; then