Skip to content

Commit 365cce6

Browse files
Bihan  RanaBihan  Rana
authored andcommitted
Fix multiple jobs assigned to same instance
1 parent 8438716 commit 365cce6

1 file changed

Lines changed: 9 additions & 8 deletions

File tree

src/dstack/_internal/server/background/scheduled_tasks/submitted_jobs.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -544,14 +544,15 @@ async def _find_assignment_fleet_with_offers(
544544
await exit_stack.enter_async_context(
545545
get_locker(get_db().dialect_name).lock_ctx(InstanceModel.__tablename__, instances_ids)
546546
)
547-
fleets_with_instances_ids = [f.id for f in fleet_models_with_instances]
548-
fleet_models_with_instances = await _refetch_fleet_models_with_instances(
549-
session=session,
550-
fleets_ids=fleets_with_instances_ids,
551-
instances_ids=instances_ids,
552-
fleet_filters=fleet_filters,
553-
instance_filters=instance_filters,
554-
)
547+
if is_db_sqlite():
548+
fleets_with_instances_ids = [f.id for f in fleet_models_with_instances]
549+
fleet_models_with_instances = await _refetch_fleet_models_with_instances(
550+
session=session,
551+
fleets_ids=fleets_with_instances_ids,
552+
instances_ids=instances_ids,
553+
fleet_filters=fleet_filters,
554+
instance_filters=instance_filters,
555+
)
555556
fleet_models = fleet_models_with_instances + fleet_models_without_instances
556557
fleet_model, fleet_instances_with_offers, _ = await find_optimal_fleet_with_offers(
557558
project=context.project,

0 commit comments

Comments
 (0)