From 5417318292d60b6806759e36de08e69dd9caf005 Mon Sep 17 00:00:00 2001
From: JunhaoQiu <56094690+qchiujunhao@users.noreply.github.com>
Date: Fri, 13 Dec 2024 11:04:21 -0500
Subject: [PATCH] added single quotes to parms

---
 tools/pycaret_predict.xml |  4 ++--
 tools/pycaret_train.xml   | 14 +++++++-------
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/tools/pycaret_predict.xml b/tools/pycaret_predict.xml
index fa91622..b3ef150 100644
--- a/tools/pycaret_predict.xml
+++ b/tools/pycaret_predict.xml
@@ -7,9 +7,9 @@
     <command>
         <![CDATA[
         echo $target_feature && 
-        python $__tool_directory__/pycaret_predict.py --model_path $input_model --data_path $input_dataset --task $model_type 
+        python $__tool_directory__/pycaret_predict.py --model_path '$input_model' --data_path '$input_dataset' --task '$model_type' 
         #if $target_feature
-            --target $target_feature
+            --target '$target_feature'
         #end if
         ]]>
     </command>
diff --git a/tools/pycaret_train.xml b/tools/pycaret_train.xml
index ed7b221..a4376e7 100644
--- a/tools/pycaret_train.xml
+++ b/tools/pycaret_train.xml
@@ -6,20 +6,20 @@
     <expand macro="python_requirements" />
     <command>
         <![CDATA[
-        python $__tool_directory__/pycaret_train.py --input_file $input_file --target_col $target_feature --output_dir . --random_seed $random_seed
+        python $__tool_directory__/pycaret_train.py --input_file '$input_file' --target_col '$target_feature' --output_dir '.' --random_seed '$random_seed'
         #if $model_type == "classification"
             #if $classification_models
-                --models $classification_models
+                --models '$classification_models'
             #end if
         #end if
         #if $model_type == "regression"
             #if $regression_models
-                --models $regression_models
+                --models '$regression_models'
             #end if
         #end if
         #if $customize_defaults == "true"
                 #if $train_size
-                --train_size $train_size 
+                --train_size '$train_size' 
                 #end if
                 #if $normalize
                 --normalize  
@@ -31,7 +31,7 @@
                 --cross_validation 
                 #end if
                 #if $cross_validation_folds
-                --cross_validation_folds $cross_validation_folds 
+                --cross_validation_folds '$cross_validation_folds'
                 #end if
                 #if $remove_outliers
                 --remove_outliers  
@@ -47,9 +47,9 @@
                 #end if
         #end if
         #if $test_file
-            --test_file $test_file 
+            --test_file '$test_file'
         #end if 
-        --model_type $model_type    
+        --model_type '$model_type'    
         ]]>
     </command>
     <inputs>