Fix static checks

astronomer · Nov 28, 2024 · 866cbc9 · 866cbc9
1 parent 2d7cdc2
commit 866cbc9
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 6 deletions.
diff --git a/ray_provider/constants.py b/ray_provider/constants.py
@@ -1,4 +1,3 @@
 from ray.job_submission import JobStatus
 
 TERMINAL_JOB_STATUSES = {JobStatus.SUCCEEDED, JobStatus.STOPPED, JobStatus.FAILED}
-
diff --git a/tests/test_triggers.py b/tests/test_triggers.py
@@ -43,7 +43,10 @@ async def test_run_no_job_id(self, mock_hook, mock_job_status):
         )
 
     @pytest.mark.asyncio
-    @patch("ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status", side_effect=[JobStatus.RUNNING, JobStatus.SUCCEEDED])
+    @patch(
+        "ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status",
+        side_effect=[JobStatus.RUNNING, JobStatus.SUCCEEDED],
+    )
     @patch("ray_provider.triggers.RayJobTrigger.hook")
     async def test_run_job_succeeded(self, mock_hook, mock_job_status):
         trigger = RayJobTrigger(
@@ -65,7 +68,10 @@ async def test_run_job_succeeded(self, mock_hook, mock_job_status):
         )
 
     @pytest.mark.asyncio
-    @patch("ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status", side_effect=[JobStatus.RUNNING, JobStatus.STOPPED])
+    @patch(
+        "ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status",
+        side_effect=[JobStatus.RUNNING, JobStatus.STOPPED],
+    )
     @patch("ray_provider.triggers.RayJobTrigger.hook")
     async def test_run_job_stopped(self, mock_hook, mock_job_status, trigger):
         generator = trigger.run()
@@ -80,7 +86,9 @@ async def test_run_job_stopped(self, mock_hook, mock_job_status, trigger):
         )
 
     @pytest.mark.asyncio
-    @patch("ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status", side_effect=[JobStatus.RUNNING, JobStatus.FAILED])
+    @patch(
+        "ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status", side_effect=[JobStatus.RUNNING, JobStatus.FAILED]
+    )
     @patch("ray_provider.triggers.RayJobTrigger.hook")
     async def test_run_job_failed(self, mock_hook, mock_job_status, trigger):
         generator = trigger.run()
@@ -95,7 +103,10 @@ async def test_run_job_failed(self, mock_hook, mock_job_status, trigger):
         )
 
     @pytest.mark.asyncio
-    @patch("ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status", side_effect=[JobStatus.RUNNING, JobStatus.SUCCEEDED])
+    @patch(
+        "ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status",
+        side_effect=[JobStatus.RUNNING, JobStatus.SUCCEEDED],
+    )
     @patch("ray_provider.triggers.RayJobTrigger.hook")
     @patch("ray_provider.triggers.RayJobTrigger._stream_logs")
     async def test_run_with_log_streaming(self, mock_stream_logs, mock_hook, mock_job_status, trigger):
@@ -180,7 +191,10 @@ async def test_cleanup_without_cluster_yaml(self, mock_log_info):
 
     @pytest.mark.asyncio
     @patch("asyncio.sleep", new_callable=AsyncMock)
-    @patch("ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status", side_effect=[JobStatus.RUNNING, JobStatus.RUNNING, JobStatus.SUCCEEDED])
+    @patch(
+        "ray_provider.triggers.RayJobTrigger.hook.get_ray_job_status",
+        side_effect=[JobStatus.RUNNING, JobStatus.RUNNING, JobStatus.SUCCEEDED],
+    )
     @patch("ray_provider.triggers.RayJobTrigger.hook")
     async def test_poll_status(self, mock_hook, mock_job_status, mock_sleep, trigger):
         await trigger._poll_status()
Original file line number	Diff line number	Diff line change
		@@ -1,4 +1,3 @@
		from ray.job_submission import JobStatus

		TERMINAL_JOB_STATUSES = {JobStatus.SUCCEEDED, JobStatus.STOPPED, JobStatus.FAILED}