update action based on pycaret==2.0

Author: PyCaret

action.yml

@@ -1,13 +1,13 @@
 name: "PyCaret AutoML Git Action"
-description: "pycaret.classification"
+description: "A simple example of AutoML created using PyCaret 2.0"
 author: "Moez Ali"
 inputs:
   DATASET:
     description: "Dataset for Training"
     required: true
     default: "juice"
   TARGET:
-    description: "Target variable"
+    description: "Name of Target variable"
     required: true
     default: "Purchase"        
 outputs:

app.py

@@ -8,8 +8,12 @@
 data = pd.read_csv(dataset_path)
 data.head()
 
-from pycaret.classification import *
-clf1 = setup(data, target = target, session_id=123, silent=True, html=False, log_experiment=True, experiment_name='exp_github')
+if type(target) is float:
+    from pycaret.regression import *
+else:
+    from pycaret.classification import *
+
+exp1 = setup(data, target = target, session_id=123, silent=True, html=False, log_experiment=True, experiment_name='exp_github')
 
 c = compare_models()
 

requirements.txt

@@ -1 +1 @@
-pycaret-nightly==0.24
+pycaret==2.0