diff --git a/tests/integration/test_api.py b/tests/integration/test_api.py index 02391bf78..28e8304f4 100644 --- a/tests/integration/test_api.py +++ b/tests/integration/test_api.py @@ -4492,6 +4492,8 @@ async def test_create_with_custom_volumes( "uri": f"job://test-cluster/{regular_user.name}/{job_id}", "restart_policy": "never", "privileged": False, + "being_dropped": False, + "logs_removed": False, } response_payload = await jobs_client.long_polling_by_job_id( @@ -4539,6 +4541,8 @@ async def test_create_with_custom_volumes( "uri": f"job://test-cluster/{regular_user.name}/{job_id}", "restart_policy": "never", "privileged": False, + "being_dropped": False, + "logs_removed": False, } @pytest.mark.asyncio @@ -4632,6 +4636,8 @@ async def test_job_failed( "uri": f"job://test-cluster/{regular_user.name}/{job_id}", "restart_policy": "never", "privileged": False, + "being_dropped": False, + "logs_removed": False, } @pytest.mark.asyncio @@ -4737,6 +4743,8 @@ async def test_create_gpu_model( "uri": f"job://test-cluster/{regular_user.name}/{job_id}", "restart_policy": "never", "privileged": False, + "being_dropped": False, + "logs_removed": False, } @pytest.mark.asyncio @@ -4836,6 +4844,8 @@ async def test_create_tpu_model( "uri": f"job://test-cluster/{regular_user.name}/{job_id}", "restart_policy": "never", "privileged": False, + "being_dropped": False, + "logs_removed": False, } diff --git a/tests/unit/test_job.py b/tests/unit/test_job.py index e10470826..76814842f 100644 --- a/tests/unit/test_job.py +++ b/tests/unit/test_job.py @@ -879,6 +879,8 @@ def test_to_primitive( "restart_policy": "never", "privileged": False, "total_price_credits": "0", + "being_dropped": False, + "logs_removed": False, } def test_to_primitive_with_max_run_time( @@ -916,6 +918,8 @@ def test_to_primitive_with_max_run_time( "restart_policy": "never", "privileged": False, "total_price_credits": "0", + "being_dropped": False, + "logs_removed": False, } def test_to_primitive_with_tags( @@ -1238,6 +1242,8 @@ def test_to_and_from_primitive( "restart_policy": str(JobRestartPolicy.ALWAYS), "privileged": False, "total_price_credits": "0", + "being_dropped": False, + "logs_removed": False, } actual = Job.to_primitive( Job.from_primitive(mock_orchestrator.config, expected)