adding a backtrack to the FeatureSet to get inference on all the data

SuperCowPowers · Jul 7, 2024 · d7e2a3d · d7e2a3d
1 parent dd2ad5c
commit d7e2a3d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/sageworks/web_components/plugins/scatter_plot.py b/src/sageworks/web_components/plugins/scatter_plot.py
@@ -245,7 +245,9 @@ def update_graph(x_value, y_value, color_value, regression_line):
 
     # Run an integration test
     from sageworks.api import Endpoint
+    from sageworks.utils.endpoint_utils import backtrack_to_fs
 
     end = Endpoint("abalone-qr-end")
-    df = end.auto_inference()
-    PluginUnitTest(ScatterPlot, input_data=df).run()
+    fs_data = backtrack_to_fs(end).pull_dataframe()
+    pred_df = end.inference(fs_data)
+    PluginUnitTest(ScatterPlot, input_data=pred_df).run()