From c338b6dc5fbb4831073722aed4c0023dc2616167 Mon Sep 17 00:00:00 2001
From: Silviu Marian Udrescu <smu226@nyu.edu>
Date: Thu, 30 Apr 2020 17:24:48 -0400
Subject: [PATCH] Add files via upload

---
 Code/S_run_aifeynman.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Code/S_run_aifeynman.py b/Code/S_run_aifeynman.py
index 8502997..b92f1fe 100644
--- a/Code/S_run_aifeynman.py
+++ b/Code/S_run_aifeynman.py
@@ -160,6 +160,7 @@ def run_aifeynman(pathdir,filename,BF_try_time,BF_ops_file_type, polyfit_deg=4,
     if test_data.size != 0:
         np.savetxt(pathdir+filename+"_test",test_data)
 
+    PA = ParetoSet()
     # Run the code on the train data 
     PA = run_AI_all(pathdir,filename+"_train",BF_try_time,BF_ops_file_type, polyfit_deg, NN_epochs)
     PA_list = PA.get_pareto_points()