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 1, 2024
1 parent e141954 commit 1942aa5
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions scripts/utils/trt_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -224,7 +224,7 @@ def try_set_inputs():

left = ctx.infer_shapes()
assert len(left) == 0

def infer(self, stream, use_cuda_graph=False):
if use_cuda_graph:
if self.cuda_graph_instance is not None:
Expand Down Expand Up @@ -433,7 +433,7 @@ def forward_trt_runner(self, trt_inputs):
ret = list(ret.values())
ret = [r.cuda() for r in ret]
# check = [check_m(r) for r in ret]
if len(ret)==1:
if len(ret) == 1:
ret = ret[0]
return ret

Expand Down
2 changes: 1 addition & 1 deletion vista3d/modeling/segresnetds.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
from __future__ import annotations

from collections.abc import Callable
from typing import Union, List, Tuple
from typing import List, Tuple, Union

import numpy as np
import torch
Expand Down

0 comments on commit 1942aa5

Please sign in to comment.