[Cherry-Pick-Main][Server] Feature table not being returned with pipeline async results (#47)

mkaliberda · web-flow · commit 8b94c17c3cbe · 2025-01-15T17:16:47.000-08:00
diff --git a/func_tests/pytest_tests/python_core_functions/test_functions.py b/func_tests/pytest_tests/python_core_functions/test_functions.py
@@ -196,9 +196,7 @@ def test_dsk_pipeline_data(dsk_proj):
         },
     ]
     data, stats = dsk.pipeline.get_cached_data(pipeline_step=1)
-    import pdb
 
-    pdb.set_trace()
     assert expected_datasegment == data
 
 
diff --git a/src/server/datamanager/sandbox.py b/src/server/datamanager/sandbox.py
@@ -754,6 +754,7 @@ def async_retrieve(self, request, project_uuid, sandbox_uuid, err_queue=deque())
                     statistics_summary = calculate_feature_stats(
                         feature_data, feature_table, label_column, sandbox.uuid
                     )
+                    summary = feature_data
 
                 elif sandbox.result_type == "grid_search":
                     result_name = "grid_result.{}".format(sandbox_uuid)

Original file line number	Diff line number	Diff line change
`@@ -196,9 +196,7 @@ def test_dsk_pipeline_data(dsk_proj):`
`196`	`196`	`},`
`197`	`197`	`]`
`198`	`198`	`data, stats = dsk.pipeline.get_cached_data(pipeline_step=1)`
`199`		`- import pdb`
`200`	`199`
`201`		`- pdb.set_trace()`
`202`	`200`	`assert expected_datasegment == data`
`203`	`201`
`204`	`202`
Original file line number	Diff line number	Diff line change
`@@ -754,6 +754,7 @@ def async_retrieve(self, request, project_uuid, sandbox_uuid, err_queue=deque())`
`754`	`754`	`statistics_summary = calculate_feature_stats(`
`755`	`755`	`feature_data, feature_table, label_column, sandbox.uuid`
`756`	`756`	`)`
	`757`	`+ summary = feature_data`
`757`	`758`
`758`	`759`	`elif sandbox.result_type == "grid_search":`
`759`	`760`	`result_name = "grid_result.{}".format(sandbox_uuid)`