From 46d40c4452c709bd8b7da0e9d42cd72cd8290412 Mon Sep 17 00:00:00 2001
From: Andrei Vishniakov <31008759+avishniakov@users.noreply.github.com>
Date: Tue, 9 Apr 2024 17:40:00 +0200
Subject: [PATCH] Fix mariadb test script (#2599) (#2600)

* fix mariadb script

* restore ci state
---
 scripts/test-migrations-mariadb.sh | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/scripts/test-migrations-mariadb.sh b/scripts/test-migrations-mariadb.sh
index 10d63dedbbd..e5329ab20bc 100755
--- a/scripts/test-migrations-mariadb.sh
+++ b/scripts/test-migrations-mariadb.sh
@@ -64,6 +64,8 @@ do
 
     git checkout release/$VERSION
     uv pip install -e ".[templates,server]"
+    # handles copier conflicts
+    uv pip install "pyyaml-include<2.0"
 
     export ZENML_ANALYTICS_OPT_IN=false
     export ZENML_DEBUG=true
@@ -87,7 +89,8 @@ source ".venv-current-branch/bin/activate"
 
 uv pip install -U setuptools wheel pip
 uv pip install -e ".[templates,server]"
-uv pip install importlib_metadata
+# handles copier conflicts
+uv pip install importlib_metadata "pyyaml-include<2.0"
 
 zenml connect --url mysql://127.0.0.1/zenml --username root --password password
 
@@ -161,6 +164,8 @@ for i in "${!MIGRATION_VERSIONS[@]}"; do
 
     git checkout release/${MIGRATION_VERSIONS[$i]}
     uv pip install -e ".[templates,server]"
+    # handles copier conflicts
+    uv pip install "pyyaml-include<2.0"
 
     export ZENML_ANALYTICS_OPT_IN=false
     export ZENML_DEBUG=true