Add additional versions to GitHub Actions tests

TylerYep · Nov 4, 2022 · 2bd59a1 · 2bd59a1
1 parent 15a7444
commit 2bd59a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/torchinfo_test.py b/tests/torchinfo_test.py
@@ -536,7 +536,7 @@ def test_parameters_with_other_layers() -> None:
 
 def test_nested_leftover_params() -> None:
     x = torch.zeros(100, 2)
-    result = summary(InsideModel(), input_data=(x,), row_settings=("var_names",))
+    result = summary(InsideModel(), input_data=[x], row_settings=("var_names",))
     expected = sum(p.numel() for p in InsideModel().parameters() if p.requires_grad)
     assert result.total_params == expected == 8