Skip to content

Commit 67657f8

Browse files
junjiang-labcopybara-github
authored andcommitted
Recover unit tests for tfl.slice and strided_slice.
PiperOrigin-RevId: 810510878
1 parent 8c8236b commit 67657f8

File tree

1 file changed

+0
-8
lines changed

1 file changed

+0
-8
lines changed

ai_edge_torch/_convert/test/test_convert.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -636,10 +636,6 @@ def forward(self, x: torch.Tensor, y: torch.Tensor):
636636
self.fail(f"Conversion failed with 6d inputs: {err}")
637637
# pylint: enable=broad-except
638638

639-
@googletest.skipIf(
640-
ai_edge_torch.config.in_oss,
641-
reason="wait until dependencies are released to tf-nightly",
642-
)
643639
def test_convert_model_with_slice_6d_inputs(self):
644640
"""Test converting a simple model with slice and 6d inputs."""
645641

@@ -658,10 +654,6 @@ def forward(self, x: torch.Tensor):
658654
self.fail(f"Conversion failed with 6d inputs for slice: {err}")
659655
# pylint: enable=broad-except
660656

661-
@googletest.skipIf(
662-
ai_edge_torch.config.in_oss,
663-
reason="wait until dependencies are released to tf-nightly",
664-
)
665657
def test_convert_model_with_strided_slice_6d_inputs(self):
666658
"""Test converting a simple model with strided_slice and 6d inputs."""
667659

0 commit comments

Comments
 (0)