Skip to content

Commit a1d4325

Browse files
committed
Addressing code review comments
Signed-off-by: Boris Fomitchev <bfomitchev@nvidia.com>
1 parent 61ad578 commit a1d4325

1 file changed

Lines changed: 11 additions & 14 deletions

File tree

nemo/export/tensorrt_lazy_compiler.py

Lines changed: 11 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -285,20 +285,17 @@ def unroll_input(input_names, input_example):
285285

286286
def unroll_one(name, val):
287287
res = {}
288-
try:
289-
if val is not None:
290-
if isinstance(val, dict):
291-
for key, data in val.items():
292-
subname = f"{name}_{key}"
293-
vals = unroll_one(subname, data)
294-
res.update(vals)
295-
elif isinstance(val, list) or isinstance(val, tuple):
296-
for i in range(len(val)):
297-
res.update(unroll_one(f"{name}_{i}", val[i]))
298-
else:
299-
res[name] = make_tensor(val)
300-
except Exception:
301-
pass
288+
if val is not None:
289+
if isinstance(val, dict):
290+
for key, data in val.items():
291+
subname = f"{name}_{key}"
292+
vals = unroll_one(subname, data)
293+
res.update(vals)
294+
elif isinstance(val, list) or isinstance(val, tuple):
295+
for i in range(len(val)):
296+
res.update(unroll_one(f"{name}_{i}", val[i]))
297+
else:
298+
res[name] = make_tensor(val)
302299
return res
303300

304301
unrolled_input = {}

0 commit comments

Comments
 (0)