fixed smoketests

Author: sid-rl

tests/smoketests/sdk/test_async_benchmark_run.py

@@ -106,8 +106,8 @@ async def test_benchmark_run_create_and_cancel(self, async_sdk_client: AsyncRunl
 
         # Start a new benchmark run
         run_data = await async_sdk_client.api.benchmarks.start_run(
-            benchmark.id,
-            name="sdk-smoketest-async-benchmark-run",
+            benchmark_id=benchmark.id,
+            run_name="sdk-smoketest-async-benchmark-run",
         )
 
         try:
@@ -123,7 +123,7 @@ async def test_benchmark_run_create_and_cancel(self, async_sdk_client: AsyncRunl
             # Get info
             info = await benchmark_run.get_info()
             assert info.id == run_data.id
-            assert info.state in ["queued", "running", "completed", "canceled"]
+            assert info.state in ["running", "completed", "canceled"]
 
             # Cancel the run
             result = await benchmark_run.cancel()

tests/smoketests/sdk/test_benchmark_run.py

@@ -106,8 +106,8 @@ def test_benchmark_run_create_and_cancel(self, sdk_client: RunloopSDK) -> None:
 
         # Start a new benchmark run
         run_data = sdk_client.api.benchmarks.start_run(
-            benchmark.id,
-            name="sdk-smoketest-benchmark-run",
+            benchmark_id=benchmark.id,
+            run_name="sdk-smoketest-benchmark-run",
         )
 
         try:
@@ -123,7 +123,7 @@ def test_benchmark_run_create_and_cancel(self, sdk_client: RunloopSDK) -> None:
             # Get info
             info = benchmark_run.get_info()
             assert info.id == run_data.id
-            assert info.state in ["queued", "running", "completed", "canceled"]
+            assert info.state in ["running", "completed", "canceled"]
 
             # Cancel the run
             result = benchmark_run.cancel()