braindecode 0.4.6__tar.gz → 0.4.7__tar.gz

{Braindecode-0.4.6 → Braindecode-0.4.7}/Braindecode.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: Braindecode
-Version: 0.4.6
+Version: 0.4.7
 Summary: A deep learning toolbox to decode raw time-domain EEG.
 Home-page: https://github.com/robintibor/braindecode
 Author: Robin Tibor Schirrmeister

{Braindecode-0.4.6 → Braindecode-0.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.1
 Name: Braindecode
-Version: 0.4.6
+Version: 0.4.7
 Summary: A deep learning toolbox to decode raw time-domain EEG.
 Home-page: https://github.com/robintibor/braindecode
 Author: Robin Tibor Schirrmeister

{Braindecode-0.4.6 → Braindecode-0.4.7}/braindecode/datautil/splitters.py

@@ -179,7 +179,7 @@ def split_into_train_test(dataset, n_folds, i_test_fold, rng=None):
     folds = get_balanced_batches(n_trials, rng, shuffle,
                                  n_batches=n_folds)
     test_inds = folds[i_test_fold]
-    all_inds = list(range(len(n_trials)))
+    all_inds = list(range(n_trials))
     train_inds = np.setdiff1d(all_inds, test_inds)
     assert np.intersect1d(train_inds, test_inds).size == 0
     assert np.array_equal(np.sort(np.union1d(train_inds, test_inds)),

{Braindecode-0.4.6 → Braindecode-0.4.7}/braindecode/torch_ext/modules.py

@@ -28,8 +28,10 @@ class Expression(torch.nn.Module):
                 expression_str = "{:s} {:s}".format(
                     self.expression_fn.func.__name__,
                     str(self.expression_fn.kwargs))
-        else:
+        elif hasattr(self.expression_fn, '__name__'):
             expression_str = self.expression_fn.__name__
+        else:
+            expression_str = repr(self.expression_fn)
         return (self.__class__.__name__ + '(' +
                 'expression=' + str(expression_str) + ')')
 
@@ -46,12 +48,15 @@ class AvgPool2dWithConv(torch.nn.Module):
         Stride of the pooling operation.
     dilation: int or (int,int)
         Dilation applied to the pooling filter.
+    padding: int or (int,int)
+        Padding applied before the pooling operation.
     """
-    def __init__(self, kernel_size, stride, dilation=1):
+    def __init__(self, kernel_size, stride, dilation=1, padding=0):
         super(AvgPool2dWithConv, self).__init__()
         self.kernel_size = kernel_size
         self.stride = stride
         self.dilation = dilation
+        self.padding = padding
         # don't name them "weights" to
         # make sure these are not accidentally used by some procedure
         # that initializes parameters or something
@@ -79,6 +84,7 @@ class AvgPool2dWithConv(torch.nn.Module):
 
         pooled = F.conv2d(x, self._pool_weights, bias=None, stride=self.stride,
                           dilation=self.dilation,
+                          padding=self.padding,
                           groups=in_channels,)
         return pooled
 

Braindecode-0.4.7/braindecode/version.py

@@ -0,0 +1 @@
+__version__ = "0.4.7"

Braindecode-0.4.6/braindecode/version.py

@@ -1 +0,0 @@
-__version__ = "0.4.6"