2d 3d tuple fix

ThijsKuipers1995 · Aug 20, 2023 · 5bdd13d · 5bdd13d
1 parent 51c68ae
commit 5bdd13d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/gconv/gnn/modules/gconv.py b/gconv/gnn/modules/gconv.py
@@ -102,13 +102,13 @@ def __init__(
         if conv_mode == "2d":
             self._conv_forward = self._conv2d_forward
             bias_shape = (1, 1, 1)
-            self.padding = (
-                _triple(padding) if isinstance(self.padding, int) else padding
-            )
+            self.padding = _pair(padding) if isinstance(self.padding, int) else padding
         elif conv_mode == "3d":
             self._conv_forward = self._conv3d_forward
             bias_shape = (1, 1, 1, 1)
-            self.padding = _pair(padding) if isinstance(self.padding, int) else padding
+            self.padding = (
+                _triple(padding) if isinstance(self.padding, int) else padding
+            )
 
         else:
             raise ValueError(