xcom obj

Tauffer-Consulting · Feb 12, 2024 · e69dd77 · e69dd77
1 parent af1e116
commit e69dd77
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/domino/base_piece.py b/src/domino/base_piece.py
@@ -301,6 +301,8 @@ def run_piece_function(
 
         # Push XCom
         xcom_obj = self.format_xcom(output_obj=output_obj)
+        self._storage_size_in_bytes = self._get_folder_size(base_results_path)
+        xcom_obj['_storage_size_in_bytes'] = self._storage_size_in_bytes
         self.push_xcom(xcom_obj=xcom_obj)
         base_results_path = f"{self.workflow_shared_storage_path}/{self.task_id}"
         self._storage_size_in_bytes = self._get_folder_size(base_results_path)