Skip to content

Commit 47e51d3

Browse files
authored
chore: better logging when insufficient resources (#271)
Signed-off-by: Terry Kong <terryk@nvidia.com>
1 parent 98473c6 commit 47e51d3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

nemo_reinforcer/distributed/virtual_cluster.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,8 @@ def __init__(
160160
self._init_placement_groups(placement_group_strategy)
161161
# Reaching here means we were successful
162162
break
163-
except ResourceInsufficientError:
163+
except ResourceInsufficientError as e:
164+
print(e)
164165
print(
165166
f"Retrying placement group creation... {i + 1}/{max_retries}. Next retry in {2**i} seconds."
166167
)

0 commit comments

Comments
 (0)