Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Aug 18, 2024
1 parent 2ec59eb commit 2047da0
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions scripts/infer.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,9 +33,7 @@
from .train import CONFIG
from .utils.trans_utils import VistaPostTransform, get_largest_connected_component_point

trt_wrap, TRT_AVAILABLE = optional_import(
"monai.networks", name="trt_wrap"
)
trt_wrap, TRT_AVAILABLE = optional_import("monai.networks", name="trt_wrap")

rearrange, _ = optional_import("einops", name="rearrange")
sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
Expand Down Expand Up @@ -143,7 +141,7 @@ def __init__(self, config_file="./configs/infer.yaml", **override):
"builder_optimization_level": 5,
"precision_constraints": "obey",
},
"timestamp": ts
"timestamp": ts,
}

trt_wrap(
Expand Down

0 comments on commit 2047da0

Please sign in to comment.