From 49a89de3afa5dac0dfdc689c5cec84ab842aa20e Mon Sep 17 00:00:00 2001 From: Philip Montgomery Date: Mon, 5 Feb 2024 11:24:14 -0500 Subject: [PATCH] fix for pyright errors --- cli/sparklespray/cluster_service.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/cli/sparklespray/cluster_service.py b/cli/sparklespray/cluster_service.py index 627b5be..7cff740 100644 --- a/cli/sparklespray/cluster_service.py +++ b/cli/sparklespray/cluster_service.py @@ -226,6 +226,7 @@ def cleanup_node_reqs(self, job_id): # before updating node requests, make sure the state has been updated to # reflect which ones are actually still running and which are done. # node_req_store.cleanup_cluster() assumes the states are up to date. + state = self.get_state(job_id) state.update() @@ -306,6 +307,7 @@ def test_image( prev_event_description = None while True: status = self.nodes.get_add_node_status(operation_id) + assert status is not None event_description = status.last_event_description if event_description != prev_event_description: print(f"Latest event: {event_description}")