def call(self, x):
                    """
                    Call function to evaluate the model.

                    :param x: Input to the model
                    :return: Prediction of the model
                    """
                    x = self.conv1(x)
                    x = self.maxpool(x)
                    x = self.conv2(x)
                    x = self.maxpool(x)
                    x = self.flatten(x)
                    x = self.dense1(x)
                    x = self.logits(x)
                    return x

                    Call function to evaluate the model.

                    :param x: Input to the model
                    :return: Prediction of the model
                    
tests/estimators/classification/test_deep_partition_ensemble.py