Skip to content

Commit

Permalink
Use contextlib's suppress method to silence an error (#8668)
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Jul 21, 2022
1 parent 602d7ff commit 4c1784b
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions models/yolo.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
"""

import argparse
import contextlib
import os
import platform
import sys
Expand Down Expand Up @@ -259,10 +260,8 @@ def parse_model(d, ch): # model_dict, input_channels(3)
for i, (f, n, m, args) in enumerate(d['backbone'] + d['head']): # from, number, module, args
m = eval(m) if isinstance(m, str) else m # eval strings
for j, a in enumerate(args):
try:
with contextlib.suppress(NameError):
args[j] = eval(a) if isinstance(a, str) else a # eval strings
except NameError:
pass

n = n_ = max(round(n * gd), 1) if n > 1 else n # depth gain
if m in (Conv, GhostConv, Bottleneck, GhostBottleneck, SPP, SPPF, DWConv, MixConv2d, Focus, CrossConv,
Expand Down

0 comments on commit 4c1784b

Please sign in to comment.