Skip to content

Commit

Permalink
Hot-Fix hang issue in integration test (#3815)
Browse files Browse the repository at this point in the history
Fix import structure
  • Loading branch information
harimkang authored Aug 8, 2024
1 parent 59a4223 commit df498b7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/otx/core/data/transform_libs/torchvision.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,6 @@
from torchvision._utils import sequence_to_str
from torchvision.transforms.v2 import functional as F # noqa: N812

from otx.algo.keypoint_detection.utils.simcc_label import SimCCLabel
from otx.core.data.entity.action_classification import ActionClsDataEntity
from otx.core.data.entity.base import (
OTXDataEntity,
Expand Down Expand Up @@ -3521,6 +3520,7 @@ def __init__(
is_numpy_to_tvtensor: bool = False,
) -> None:
super().__init__()
from otx.algo.keypoint_detection.utils.simcc_label import SimCCLabel

self.encoder = SimCCLabel(
input_size=(192, 256),
Expand Down

0 comments on commit df498b7

Please sign in to comment.