remove prints

Author: christinadionysio

src/main/python/systemds/scuro/drsearch/node_executor.py

@@ -497,14 +497,14 @@ def _checkpoint_memory_usage(
                 print(
                     f"UNDERESTIMATED GPU PEAK MEMORY: GPU peak bytes: {gpu_peak_bytes/1024**3:.2f} GB, Estimated GPU bytes: {self.scheduler.node_resources[node_id][1]/1024**3:.2f} GB for node {node_id}: {operation_name}"
                 )
-        if self.scheduler.node_resources[node_id][0] >= peak_bytes * 2:
-            print(
-                f"Peak bytes: {peak_bytes/1024**3:.2f} GB, Estimated CPU bytes: {self.scheduler.node_resources[node_id][0]/1024**3:.2f} GB, 200% of estimated for node {node_id}: {operation_name}"
-            )
-        if self.scheduler.node_resources[node_id][1] > gpu_peak_bytes * 2:
-            print(
-                f"GPU peak bytes: {gpu_peak_bytes/1024**3:.2f} GB, Estimated GPU bytes: {self.scheduler.node_resources[node_id][1]/1024**3:.2f} GB, 200% of estimated for node {node_id}: {operation_name}"
-            )
+            if self.scheduler.node_resources[node_id][0] >= peak_bytes * 2:
+                print(
+                    f"Peak bytes: {peak_bytes/1024**3:.2f} GB, Estimated CPU bytes: {self.scheduler.node_resources[node_id][0]/1024**3:.2f} GB, 200% of estimated for node {node_id}: {operation_name}"
+                )
+            if self.scheduler.node_resources[node_id][1] > gpu_peak_bytes * 2:
+                print(
+                    f"GPU peak bytes: {gpu_peak_bytes/1024**3:.2f} GB, Estimated GPU bytes: {self.scheduler.node_resources[node_id][1]/1024**3:.2f} GB, 200% of estimated for node {node_id}: {operation_name}"
+                )
         data[node_id] = {
             "cpu_peak_bytes": peak_bytes,
             "gpu_peak_bytes": gpu_peak_bytes,

src/main/python/systemds/scuro/representations/aggregated_representation.py

@@ -94,7 +94,6 @@ def transform(self, modality):
         )
 
         aggregate_dim = (0,)
-        print("Aggregated representation transform start")
         if self.target_dimensions is not None:
             input_dimensions = self._get_input_dimensions(modality.data)