fix(python-client): improve get_job_status for running jobs

Author: rubenfiszel

python-client/wmill/wmill/client.py

@@ -257,10 +257,7 @@ def get_job_status(self, job_id: str) -> JobStatus:
         assert job_type, f"{job} is not a valid job"
         if job_type.lower() == "completedjob":
             return "COMPLETED"
-        additional_properties = job.get("additional_properties", {})
-        if "running" not in additional_properties:
-            raise Exception(f"{job_id} is not running")
-        if additional_properties.get("running"):
+        if job.get("running"):
             return "RUNNING"
         return "WAITING"