Braindecode 1.6.1.dev1053__tar.gz → 1.6.1.dev1054__tar.gz

{braindecode-1.6.1.dev1053/braindecode.egg-info → braindecode-1.6.1.dev1054}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.6.1.dev1053
+Version: 1.6.1.dev1054
 Summary: Deep learning software to decode EEG, ECG or MEG signals
 Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Alexandre Gramfort <agramfort@meta.com>
 Maintainer-email: Alexandre Gramfort <agramfort@meta.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Robin Tibor Schirrmeister <robintibor@gmail.com>

{braindecode-1.6.1.dev1053 → braindecode-1.6.1.dev1054}/braindecode/eegneuralnet.py

@@ -7,7 +7,6 @@
 import abc
 import inspect
 import logging
-import warnings
 from typing import Literal
 
 import mne
@@ -42,7 +41,6 @@ def _get_model(model: str):
 
 class _EEGNeuralNet(NeuralNet, abc.ABC):
     signal_args_set_ = False
-    _warned_drop_last_ = False
 
     @property
     def log(self):
@@ -269,44 +267,6 @@ class _EEGNeuralNet(NeuralNet, abc.ABC):
             X = X.get_data(units="uV")
         return super().get_dataset(X, y)
 
-    def get_iterator(self, dataset, training=False):
-        iterator = super().get_iterator(dataset, training=training)
-        # Warn (once per fit) if drop_last would discard every training batch,
-        # e.g. a training set smaller than batch_size. This silently skips
-        # training without raising any error, which is hard to diagnose.
-        if training and not self._warned_drop_last_:
-            self._warn_if_training_batches_dropped(iterator)
-        return iterator
-
-    def _warn_if_training_batches_dropped(self, loader):
-        """Warn if ``drop_last`` discards all training batches.
-
-        With ``iterator_train__drop_last=True`` (the braindecode default), a
-        training set smaller than ``batch_size`` yields zero batches, so no
-        forward/backward pass runs and the model is left untrained without any
-        error. We inspect the resolved training ``DataLoader`` (after
-        ``train_split`` has been applied), so the count reflects the actual
-        number of training examples.
-        """
-        self._warned_drop_last_ = True
-        if not getattr(loader, "drop_last", False):
-            return
-        batch_size = getattr(loader, "batch_size", None)
-        try:
-            n_train = len(loader.dataset)
-        except TypeError:
-            return  # e.g. an IterableDataset whose length is unknown
-        if not batch_size or n_train == 0 or n_train >= batch_size:
-            return
-        warnings.warn(
-            f"The training set has {n_train} example(s), which is smaller than "
-            f"batch_size={batch_size}. With iterator_train__drop_last=True "
-            f"(the braindecode default), all training batches are dropped and "
-            f"the model is not trained. Reduce batch_size to at most {n_train}, "
-            f"or pass iterator_train__drop_last=False.",
-            UserWarning,
-        )
-
     def partial_fit(self, X, y=None, classes=None, **fit_params):
         """Fit the module.
 
@@ -406,7 +366,6 @@ class _EEGNeuralNet(NeuralNet, abc.ABC):
           the module and to the ``self.train_split`` call.
         """
         # this needs to be executed before the net is initialized:
-        self._warned_drop_last_ = False  # re-check on every fresh fit
         if not self.signal_args_set_:
             self._set_signal_args(X, y, classes=None)
             self.signal_args_set_ = True

braindecode-1.6.1.dev1054/braindecode/version.py

@@ -0,0 +1 @@
+__version__ = "1.6.1.dev1054"

{braindecode-1.6.1.dev1053 → braindecode-1.6.1.dev1054/braindecode.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.6.1.dev1053
+Version: 1.6.1.dev1054
 Summary: Deep learning software to decode EEG, ECG or MEG signals
 Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Alexandre Gramfort <agramfort@meta.com>
 Maintainer-email: Alexandre Gramfort <agramfort@meta.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Robin Tibor Schirrmeister <robintibor@gmail.com>

{braindecode-1.6.1.dev1053 → braindecode-1.6.1.dev1054}/docs/whats_new.rst

@@ -88,15 +88,6 @@ Bug fixes
   :class:`~braindecode.EEGClassifier` infer the signal dimensions from the data.
   By `Bhargav Kowshik`_
 
-- Fix :class:`~braindecode.EEGClassifier` and
-  :class:`~braindecode.EEGRegressor` silently skipping all training when the
-  training set is smaller than ``batch_size`` (with the default
-  ``iterator_train__drop_last=True``). Previously every batch was dropped and
-  the model was left untrained without any message; a :class:`UserWarning` is
-  now raised telling the user to lower ``batch_size`` or set
-  ``iterator_train__drop_last=False``.
-  (:gh:`1053` by `Bhargav Kowshik`_)
-
 Code health
 ============
 

braindecode-1.6.1.dev1053/braindecode/version.py

@@ -1 +0,0 @@
-__version__ = "1.6.1.dev1053"