Skip to content

Commit

Permalink
removed import (#1782)
Browse files Browse the repository at this point in the history
  • Loading branch information
shaydeci authored Jan 24, 2024
1 parent 5ab9fe7 commit 9156e4a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/super_gradients/inference/iterate_over_predictions.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
from typing import Iterable, Tuple, TypeAlias, Union
from typing import Iterable, Tuple, Union

import numpy as np
import torch

__all__ = ["iterate_over_detection_predictions_in_batched_format", "iterate_over_detection_predictions_in_flat_format"]


NumpyArrayOrTensor: TypeAlias = Union[np.ndarray, torch.Tensor]
NumpyArrayOrTensor = Union[np.ndarray, torch.Tensor]


def iterate_over_detection_predictions_in_flat_format(predictions: NumpyArrayOrTensor, batch_size: int):
Expand Down

0 comments on commit 9156e4a

Please sign in to comment.