fix(result): bug in combine_results method

ImperialCollegeLondon · Jan 2, 2025 · f1c40a0 · f1c40a0
1 parent 62a0e5f
commit f1c40a0
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pyprobe/result.py b/pyprobe/result.py
@@ -600,7 +600,5 @@ def combine_results(
             pl.lit(result.info[key]).alias(key) for key in result.info.keys()
         ]
         result.live_dataframe = result.live_dataframe.with_columns(instructions)
-    combined_result = results[0].clean_copy()
-    combined_result.info = {}
-    combined_result.extend(results[0:], concat_method=concat_method)
-    return combined_result
+    results[0].extend(results[1:], concat_method=concat_method)
+    return results[0]