Skip to content

Commit 1854e3f

Browse files
authored
Fix: application timeout in fast-track if only 1 gpu available (#151)
* Fix: application timeout in fast-track if only 1 gpu * Fix: reply to message so bot 'stops thinking'
1 parent b9d5a2a commit 1854e3f

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

src/discord-cluster-manager/cogs/leaderboard_cog.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -102,9 +102,6 @@ async def select_gpu_view(
102102
"""
103103
UI displayed to user to select GPUs that they want to use.
104104
"""
105-
if not interaction.response.is_done():
106-
await interaction.response.defer(ephemeral=True)
107-
108105
view = GPUSelectionView(gpus)
109106

110107
await send_discord_message(
@@ -200,7 +197,14 @@ async def on_submit_hook(
200197

201198
# if there is more than one candidate GPU, display UI to let user select,
202199
# otherwise just run on that GPU
200+
if not interaction.response.is_done():
201+
await interaction.response.defer(ephemeral=True)
203202
if len(gpus) == 1:
203+
await send_discord_message(
204+
interaction,
205+
f"Running on GPU: **{gpus[0]}**",
206+
ephemeral=True,
207+
)
204208
selected_gpus = gpus
205209
else:
206210
view = await self.select_gpu_view(interaction, leaderboard_name, gpus)

0 commit comments

Comments
 (0)