move the completed to correct place

Signed-off-by: Lukasz Cmielowski <lcmielow@redhat.com>
Assisted-by: Cursor

Author: LukaszCmielowski

components/data_processing/automl/timeseries_data_loader/component.py

@@ -388,13 +388,6 @@ def _concat_sorted(parts: list, sort_by: list) -> pd.DataFrame:
                 "each series has enough train rows for the selection segment."
             )
 
-        status.record(
-            "split_and_export",
-            "completed",
-            test_size=test_size,
-            selection_train_size=selection_train_size,
-        )
-
         # Save test dataset to artifact
         test_df.to_csv(sampled_test_dataset.path, index=False)
 
@@ -404,6 +397,13 @@ def _concat_sorted(parts: list, sort_by: list) -> pd.DataFrame:
         selection_train_df.to_csv(selection_path, index=False)
         extra_train_df.to_csv(extra_path, index=False)
 
+        status.record(
+            "split_and_export",
+            "completed",
+            test_size=test_size,
+            selection_train_size=selection_train_size,
+        )
+
         logger.info(
             "Timeseries loader: %s rows from s3://%s/%s; split selection=%s extra=%s test=%s",
             len(df),
@@ -422,9 +422,6 @@ def _concat_sorted(parts: list, sort_by: list) -> pd.DataFrame:
             "selection_train_size": selection_train_size,
         }
 
-        status.record("write_outputs", "started")
-        status.record("write_outputs", "completed")
-
         # Sample rows for downstream use (ISO timestamps when supported; JSON string to avoid NaN issues)
         sample_tail = test_df.tail(min(5, len(test_df)))
         if hasattr(sample_tail, "to_dict"):