Skip to content

Commit 7c73a86

Browse files
dines-rldines-rl
committed
cp dines
1 parent f713a10 commit 7c73a86

File tree

5 files changed

+26
-12
lines changed

5 files changed

+26
-12
lines changed

‎src/runloop_api_client/resources/blueprints.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -333,6 +333,7 @@ def create_and_await_build_complete(
333333
file_mounts: Optional[Dict[str, str]] | Omit = omit,
334334
launch_parameters: Optional[LaunchParameters] | Omit = omit,
335335
metadata: Optional[Dict[str, str]] | Omit = omit,
336+
network_policy_id: Optional[str] | Omit = omit,
336337
secrets: Optional[Dict[str, str]] | Omit = omit,
337338
services: Optional[Iterable[blueprint_create_params.Service]] | Omit = omit,
338339
system_setup_commands: Optional[SequenceNotStr[str]] | Omit = omit,
@@ -372,6 +373,7 @@ def create_and_await_build_complete(
372373
file_mounts=file_mounts,
373374
launch_parameters=launch_parameters,
374375
metadata=metadata,
376+
network_policy_id=network_policy_id,
375377
secrets=secrets,
376378
services=services,
377379
system_setup_commands=system_setup_commands,
@@ -1001,6 +1003,7 @@ async def create_and_await_build_complete(
10011003
file_mounts: Optional[Dict[str, str]] | Omit = omit,
10021004
launch_parameters: Optional[LaunchParameters] | Omit = omit,
10031005
metadata: Optional[Dict[str, str]] | Omit = omit,
1006+
network_policy_id: Optional[str] | Omit = omit,
10041007
secrets: Optional[Dict[str, str]] | Omit = omit,
10051008
services: Optional[Iterable[blueprint_create_params.Service]] | Omit = omit,
10061009
system_setup_commands: Optional[SequenceNotStr[str]] | Omit = omit,
@@ -1040,6 +1043,7 @@ async def create_and_await_build_complete(
10401043
file_mounts=file_mounts,
10411044
launch_parameters=launch_parameters,
10421045
metadata=metadata,
1046+
network_policy_id=network_policy_id,
10431047
secrets=secrets,
10441048
services=services,
10451049
system_setup_commands=system_setup_commands,

‎src/runloop_api_client/sdk/async_benchmark.py‎

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@ async def start_run(
107107
benchmark_id=self._id,
108108
**params,
109109
)
110+
assert run_view.benchmark_id is not None, "benchmark_id should be set for runs created from a benchmark"
110111
return AsyncBenchmarkRun(self._client, run_view.id, run_view.benchmark_id)
111112

112113
async def add_scenarios(
@@ -157,8 +158,12 @@ async def list_runs(
157158
:return: List of async benchmark runs
158159
:rtype: List[AsyncBenchmarkRun]
159160
"""
160-
page = await self._client.benchmarks.runs.list(
161+
page = await self._client.benchmark_runs.list(
161162
benchmark_id=self._id,
162163
**params,
163164
)
164-
return [AsyncBenchmarkRun(self._client, run.id, run.benchmark_id) for run in page.runs]
165+
return [
166+
AsyncBenchmarkRun(self._client, run.id, run.benchmark_id)
167+
for run in page.runs
168+
if run.benchmark_id is not None
169+
]

‎src/runloop_api_client/sdk/async_benchmark_run.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ async def get_info(
7171
:return: Current benchmark run state info
7272
:rtype: BenchmarkRunView
7373
"""
74-
return await self._client.benchmarks.runs.retrieve(
74+
return await self._client.benchmark_runs.retrieve(
7575
self._id,
7676
**options,
7777
)
@@ -88,7 +88,7 @@ async def cancel(
8888
:return: Updated benchmark run state
8989
:rtype: BenchmarkRunView
9090
"""
91-
return await self._client.benchmarks.runs.cancel(
91+
return await self._client.benchmark_runs.cancel(
9292
self._id,
9393
**options,
9494
)
@@ -105,7 +105,7 @@ async def complete(
105105
:return: Completed benchmark run state
106106
:rtype: BenchmarkRunView
107107
"""
108-
return await self._client.benchmarks.runs.complete(
108+
return await self._client.benchmark_runs.complete(
109109
self._id,
110110
**options,
111111
)
@@ -120,7 +120,7 @@ async def list_scenario_runs(
120120
:return: List of async scenario run objects
121121
:rtype: List[AsyncScenarioRun]
122122
"""
123-
page = await self._client.benchmarks.runs.list_scenario_runs(
123+
page = await self._client.benchmark_runs.list_scenario_runs(
124124
self._id,
125125
**params,
126126
)

‎src/runloop_api_client/sdk/benchmark.py‎

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@ def start_run(
107107
benchmark_id=self._id,
108108
**params,
109109
)
110+
assert run_view.benchmark_id is not None, "benchmark_id should be set for runs created from a benchmark"
110111
return BenchmarkRun(self._client, run_view.id, run_view.benchmark_id)
111112

112113
def add_scenarios(
@@ -157,8 +158,12 @@ def list_runs(
157158
:return: List of benchmark runs
158159
:rtype: List[BenchmarkRun]
159160
"""
160-
page = self._client.benchmarks.runs.list(
161+
page = self._client.benchmark_runs.list(
161162
benchmark_id=self._id,
162163
**params,
163164
)
164-
return [BenchmarkRun(self._client, run.id, run.benchmark_id) for run in page.runs]
165+
return [
166+
BenchmarkRun(self._client, run.id, run.benchmark_id)
167+
for run in page.runs
168+
if run.benchmark_id is not None
169+
]

‎src/runloop_api_client/sdk/benchmark_run.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def get_info(
7171
:return: Current benchmark run state info
7272
:rtype: BenchmarkRunView
7373
"""
74-
return self._client.benchmarks.runs.retrieve(
74+
return self._client.benchmark_runs.retrieve(
7575
self._id,
7676
**options,
7777
)
@@ -88,7 +88,7 @@ def cancel(
8888
:return: Updated benchmark run state
8989
:rtype: BenchmarkRunView
9090
"""
91-
return self._client.benchmarks.runs.cancel(
91+
return self._client.benchmark_runs.cancel(
9292
self._id,
9393
**options,
9494
)
@@ -105,7 +105,7 @@ def complete(
105105
:return: Completed benchmark run state
106106
:rtype: BenchmarkRunView
107107
"""
108-
return self._client.benchmarks.runs.complete(
108+
return self._client.benchmark_runs.complete(
109109
self._id,
110110
**options,
111111
)
@@ -120,7 +120,7 @@ def list_scenario_runs(
120120
:return: List of scenario run objects
121121
:rtype: List[ScenarioRun]
122122
"""
123-
page = self._client.benchmarks.runs.list_scenario_runs(
123+
page = self._client.benchmark_runs.list_scenario_runs(
124124
self._id,
125125
**params,
126126
)

0 commit comments

Comments
 (0)