minor fixes

Author: APJansen

groco/layers/conv2d_transpose.py

@@ -57,7 +57,6 @@ def __init__(
 
         super().__init__(kernel_size=kernel_size, **kwargs)
         self.group_valued_input = None
-        self.group_order = self.group.order if transpose else self.subgroup.order
 
     @backup_and_restore(("kernel", "bias", "filters"))
     def call(self, inputs):

groco/layers/conv3d.py

@@ -60,7 +60,6 @@ def __init__(
         self.subgroup = self.group_transforms.subgroup
 
         self.group_valued_input = None
-        self.group_order = self.group.order if transpose else self.subgroup.order
 
     @backup_and_restore(("kernel", "bias", "filters"))
     def call(self, inputs):

groco/layers/conv3d_transpose.py

@@ -63,7 +63,6 @@ def __init__(
 
         super().__init__(kernel_size=kernel_size, **kwargs)
         self.group_valued_input = None
-        self.group_order = self.group.order if transpose else self.subgroup.order
 
     @backup_and_restore(("kernel", "bias", "filters"))
     def call(self, inputs):

tests/conv_test_base.py

@@ -128,7 +128,7 @@ def test_padding_equiv(self):
                 self.check_equivariance(conv_layer, signal_on_group)
 
     def generate_shape(self, group, output=True):
-        if type(group) == str:
+        if isinstance(group, str):
             group = self.group_dict[group]
         if group == None:
             shape = self.shape