Update src/frequenz/sdk/microgrid/_data_pipeline.py

Co-authored-by: Copilot <[email protected]>
Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

src/frequenz/sdk/microgrid/_data_pipeline.py

@@ -494,16 +494,16 @@ async def _stop(self) -> None:
             await self._data_sourcing_actor.actor.stop()
         if self._resampling_actor:
             await self._resampling_actor.actor.stop()
-        await self._battery_power_wrapper.stop()
-        await self._ev_power_wrapper.stop()
-        await self._pv_power_wrapper.stop()
-        for pool in self._battery_pool_reference_stores.values():
-            await pool.stop()
-        for evpool in self._ev_charger_pool_reference_stores.values():
-            await evpool.stop()
-        for pvpool in self._pv_pool_reference_stores.values():
-            await pvpool.stop()
 
+        # Parallelize the stop calls for power wrappers and pools
+        await asyncio.gather(
+            self._battery_power_wrapper.stop(),
+            self._ev_power_wrapper.stop(),
+            self._pv_power_wrapper.stop(),
+            *(pool.stop() for pool in self._battery_pool_reference_stores.values()),
+            *(evpool.stop() for evpool in self._ev_charger_pool_reference_stores.values()),
+            *(pvpool.stop() for pvpool in self._pv_pool_reference_stores.values()),
+        )
 
 _DATA_PIPELINE: _DataPipeline | None = None