Add cached run into testing of migrations (#3199)

* add cache into testing * add cache into testing
zenml-io · Nov 19, 2024 · 739f061 · 739f061
1 parent 865a441
commit 739f061
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/scripts/test-migrations.sh b/scripts/test-migrations.sh
@@ -141,8 +141,10 @@ function run_tests_for_version() {
     # Check if the version supports templates with arguments (> 0.52.0)
     if [ "$(version_compare "$VERSION" "0.52.0")" == ">" ]; then
         python3 run.py --feature-pipeline --training-pipeline --no-cache
+        python3 run.py --feature-pipeline --training-pipeline # run with cache
     else
         python3 run.py --no-cache
+        python3 run.py # run with cache
     fi
     # Add additional CLI tests here
     zenml version
@@ -177,6 +179,7 @@ function run_tests_for_version() {
         # Run the pipeline again to check if the restored database is working
         echo "===== Running starter template pipeline after DB restore (file dump) ====="
         python3 run.py --feature-pipeline --training-pipeline --no-cache
+        python3 run.py --feature-pipeline --training-pipeline # run with cache
 
         # For a mysql compatible database, perform a DB backup and restore using
         # the backup database
@@ -204,6 +207,7 @@ function run_tests_for_version() {
             # Run the pipeline again to check if the restored database is working
             echo "===== Running starter template pipeline after DB restore (backup database) ====="
             python3 run.py --feature-pipeline --training-pipeline --no-cache
+            python3 run.py --feature-pipeline --training-pipeline # run with cache
         fi
 
     else