From d6ad889d1fb6ccc095d1e19bab1604428e4ca89e Mon Sep 17 00:00:00 2001
From: sanderegg <35365065+sanderegg@users.noreply.github.com>
Date: Tue, 10 Dec 2024 17:01:48 +0100
Subject: [PATCH] improve

---
 .../pytest-simcore/src/pytest_simcore/helpers/autoscaling.py   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/pytest-simcore/src/pytest_simcore/helpers/autoscaling.py b/packages/pytest-simcore/src/pytest_simcore/helpers/autoscaling.py
index 4ecc889a96c..d21a436ace6 100644
--- a/packages/pytest-simcore/src/pytest_simcore/helpers/autoscaling.py
+++ b/packages/pytest-simcore/src/pytest_simcore/helpers/autoscaling.py
@@ -18,7 +18,7 @@
 
 def assert_cluster_state(
     spied_cluster_analysis: MockType, *, expected_calls: int, expected_num_machines: int
-) -> None:
+) -> Cluster:
     assert spied_cluster_analysis.call_count > 0
 
     assert isinstance(spied_cluster_analysis.spy_return, Cluster)
@@ -27,6 +27,7 @@ def assert_cluster_state(
         == expected_num_machines
     )
     print("current cluster state:", spied_cluster_analysis.spy_return)
+    return spied_cluster_analysis.spy_return
 
 
 def create_fake_association(