From fe60b1462416d7f25d67094ff3afeabd0338f96d Mon Sep 17 00:00:00 2001 From: Tasha Upchurch Date: Sat, 21 Jun 2025 21:19:59 -0600 Subject: [PATCH] Update tests/test_torchvision_nodes.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> --- tests/test_torchvision_nodes.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_torchvision_nodes.py b/tests/test_torchvision_nodes.py index bedca23..12c7408 100644 --- a/tests/test_torchvision_nodes.py +++ b/tests/test_torchvision_nodes.py @@ -23,7 +23,9 @@ def test_TorchVisionTransformStringParser(): transforms, name = ret assert name == "test" assert transforms[0][0] == "Resize" + assert transforms[0][1] == ["64", "64"] assert transforms[1][0] == "GaussianBlur" + assert transforms[1][1] == ["1"] def test_TorchVisionTransformNode(): node1 = TorchVisionTransformCompositionList()