add a grace for Jobs that may start in Unknown

Signed-off-by: Prekshi Vyas <prekshivyas@gmail.com>

Author: prekshivyas

nemo_run/core/execution/lepton.py

@@ -132,9 +132,7 @@ def move_data(self, sleep: float = 10, timeout: int = 600, poll_interval: int =
             current_job_status = current_job.status.state
 
             if count > 0:
-                if current_job_status == LeptonJobState.Completed:
-                    break
-                elif current_job_status == LeptonJobState.Failed:
+                if current_job_status == LeptonJobState.Completed or current_job_status == LeptonJobState.Failed:
                     break
                 elif current_job_status == LeptonJobState.Unknown:
                     if unknown_start_time is None: