Skip to content

Commit 0bf47c7

Browse files
committed
Get fleets from run_spec.merged_profile.fleets
1 parent 54c4a9f commit 0bf47c7

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

src/dstack/_internal/server/background/tasks/process_submitted_jobs.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -250,8 +250,8 @@ async def _process_submitted_job(session: AsyncSession, job_model: JobModel):
250250
]
251251
if run_model.fleet is not None:
252252
fleet_filters.append(FleetModel.id == run_model.fleet_id)
253-
if run_spec.configuration.fleets is not None:
254-
fleet_filters.append(FleetModel.name.in_(run_spec.configuration.fleets))
253+
if run_spec.merged_profile.fleets is not None:
254+
fleet_filters.append(FleetModel.name.in_(run_spec.merged_profile.fleets))
255255

256256
instance_filters = [
257257
InstanceModel.deleted == False,
@@ -295,7 +295,7 @@ async def _process_submitted_job(session: AsyncSession, job_model: JobModel):
295295
master_job_provisioning_data=master_job_provisioning_data,
296296
volumes=volumes,
297297
)
298-
if fleet_model is None and run_spec.configuration.fleets is not None:
298+
if fleet_model is None and run_spec.merged_profile.fleets is not None:
299299
# Run cannot create new fleets when fleets are specified
300300
logger.debug("%s: failed to use specified fleets", fmt(job_model))
301301
job_model.status = JobStatus.TERMINATING
@@ -538,7 +538,7 @@ def _find_optimal_fleet_with_offers(
538538
fleet_priority,
539539
)
540540
)
541-
if run_spec.configuration.fleets is None and all(
541+
if run_spec.merged_profile.fleets is None and all(
542542
t[2] == 0 for t in candidate_fleets_with_offers
543543
):
544544
# If fleets are not specified and no fleets have available offers, create a new fleet.

0 commit comments

Comments
 (0)