last fix

app.py

@@ -23,19 +23,19 @@ model = mr.get_model("cohort_model", version=1)
 model_dir = model.download()
 model = joblib.load(model_dir + "/xgb_.pkl")
 
-# Get feature view in a dataframe
-data, labels = feature_view.get_training_data(training_dataset_version=1)
-data["percentage_future"] = labels
 
-# Convert it to pandas datetime
-data["cohort_first_month"] = pd.to_datetime(data["cohort_first_month"])
-data["month"] = pd.to_datetime(data["month"])
+def cohort_predict(cohort_start, start_date, product):
 
-# Sort and assert
-data = data.sort_values(by=["cohort_first_product", "cohort_first_month", "month"])
+    # Get feature view in a dataframe
+    data, labels = feature_view.get_training_data(training_dataset_version=1)
+    data["percentage_future"] = labels
 
+    # Convert it to pandas datetime
+    data["cohort_first_month"] = pd.to_datetime(data["cohort_first_month"])
+    data["month"] = pd.to_datetime(data["month"])
 
-def cohort_predict(cohort_start, start_date, product):
+    # Sort and assert
+    data = data.sort_values(by=["cohort_first_product", "cohort_first_month", "month"])
     
     new_seq = generate_new_data(data, cohort_start, start_date, product, model, X_columns_to_drop, 12)
     hist_seq = get_sequence(data, cohort_start, product)