Skip to content

Commit

Permalink
chore: fix test import for test_decompositions.py
Browse files Browse the repository at this point in the history
  • Loading branch information
peri044 committed Feb 3, 2025
1 parent 9cb3244 commit ef36af8
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/py/dynamo/lowering/test_decompositions.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
import torch
import torch_tensorrt
from parameterized import parameterized
from testing_utilities import DECIMALS_OF_AGREEMENT, lower_graph_testing
from torch.testing._internal.common_utils import TestCase, run_tests

from ..testing_utilities import DECIMALS_OF_AGREEMENT, lower_graph_testing


class TestLowering(TestCase):
def test_lowering_inplace_op(self):
Expand Down

0 comments on commit ef36af8

Please sign in to comment.