braindecode 0.8.1__tar.gz → 1.1.0__tar.gz

{braindecode-0.8.1 → braindecode-1.1.0}/LICENSE.txt

@@ -1,6 +1,6 @@
 BSD 3-Clause License
 
-Copyright (c) 2017-2020 Braindecode Developers
+Copyright (c) 2017-currently Braindecode Developers
 All rights reserved.
 
 Redistribution and use in source and binary forms, with or without modification,

{braindecode-0.8.1 → braindecode-1.1.0}/MANIFEST.in

@@ -1,5 +1,8 @@
 include README.rst
 include LICENSE.txt
+include NOTICE.txt
+
+include braindecode/models/summary.csv
 
 recursive-include docs *.ipynb *.rst conf.py Makefile
 recursive-exclude docs *checkpoint.ipynb

braindecode-1.1.0/NOTICE.txt

@@ -0,0 +1,20 @@
+# BRAINDECODE Notice
+
+## Licensed Components
+
+### BSD-3-Clause Licensed Files
+
+All files within the `src/` directory are licensed under the BSD-3-Clause License.
+
+### CC BY-NC 4.0 Licensed Files
+
+The following components are licensed under the Creative Commons Attribution-NonCommercial 4.0 International License:
+
+- `braindecode/models/eegminer.py`
+
+As well as class later imported into the `braindecode.models.module` named as GeneralizedGaussianFilter.
+
+## License Links
+
+- [BSD-3-Clause License](https://opensource.org/licenses/BSD-3-Clause)
+- [CC BY-NC 4.0 License](https://creativecommons.org/licenses/by-nc/4.0/)

{braindecode-0.8.1/braindecode.egg-info → braindecode-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: braindecode
-Version: 0.8.1
+Version: 1.1.0
 Summary: Deep learning software to decode EEG, ECG or MEG signals
 Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>
 Maintainer-email: Alexandre Gramfort <agramfort@meta.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Robin Tibor Schirrmeister <robintibor@gmail.com>
@@ -15,32 +15,38 @@ Classifier: Intended Audience :: Science/Research
 Classifier: Topic :: Software Development :: Build Tools
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: License :: OSI Approved :: BSD License
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Python: >=3.8
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >3.9
 Description-Content-Type: text/x-rst
 License-File: LICENSE.txt
-Requires-Dist: mne
+License-File: NOTICE.txt
+Requires-Dist: mne>=1.10.0
+Requires-Dist: mne_bids>=0.16
 Requires-Dist: numpy
 Requires-Dist: pandas
 Requires-Dist: scipy
 Requires-Dist: matplotlib
 Requires-Dist: h5py
-Requires-Dist: skorch
-Requires-Dist: torch
+Requires-Dist: skorch~=0.14.0
+Requires-Dist: torch~=2.0
+Requires-Dist: torchaudio~=2.0
 Requires-Dist: einops
 Requires-Dist: joblib
-Requires-Dist: torchinfo
+Requires-Dist: torchinfo~=1.8
+Requires-Dist: wfdb
+Requires-Dist: h5py
+Requires-Dist: linear_attention_transformer
 Requires-Dist: docstring_inheritance
 Provides-Extra: moabb
-Requires-Dist: moabb>=1.0.0; extra == "moabb"
+Requires-Dist: moabb>=1.2.0; extra == "moabb"
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"
 Requires-Dist: pytest-cov; extra == "tests"
 Requires-Dist: codecov; extra == "tests"
 Requires-Dist: pytest_cases; extra == "tests"
+Requires-Dist: mypy; extra == "tests"
 Provides-Extra: docs
 Requires-Dist: sphinx_gallery; extra == "docs"
 Requires-Dist: sphinx_rtd_theme; extra == "docs"
@@ -53,6 +59,8 @@ Requires-Dist: sphinx_design; extra == "docs"
 Requires-Dist: lightning; extra == "docs"
 Requires-Dist: seaborn; extra == "docs"
 Requires-Dist: pre-commit; extra == "docs"
+Requires-Dist: openneuro-py; extra == "docs"
+Dynamic: license-file
 
 .. image:: https://badges.gitter.im/braindecodechat/community.svg
    :alt: Join the chat at https://gitter.im/braindecodechat/community
@@ -166,37 +174,13 @@ as well as the `MNE-Python <https://mne.tools>`_ software that is used by braind
 Licensing
 ^^^^^^^^^
 
-Braindecode is **BSD-licenced** (BSD-3-Clause):
-
-    This software is OSI Certified Open Source Software.
-    OSI Certified is a certification mark of the Open Source Initiative.
-
-    Copyright (c) 2011-2022, authors of Braindecode.
-    All rights reserved.
-
-    Redistribution and use in source and binary forms, with or without
-    modification, are permitted provided that the following conditions are met:
-
-    * Redistributions of source code must retain the above copyright notice,
-      this list of conditions and the following disclaimer.
+This project is primarily licensed under the BSD-3-Clause License.
 
-    * Redistributions in binary form must reproduce the above copyright notice,
-      this list of conditions and the following disclaimer in the documentation
-      and/or other materials provided with the distribution.
+Additional Components
+~~~~~~~~~~~~~~~~~~~~~
 
-    * Neither the names of braindecode authors nor the names of any
-      contributors may be used to endorse or promote products derived from
-      this software without specific prior written permission.
+Some components within this repository are licensed under the Creative Commons Attribution-NonCommercial 4.0 International
+License.
 
-    **This software is provided by the copyright holders and contributors
-    "as is" and any express or implied warranties, including, but not
-    limited to, the implied warranties of merchantability and fitness for
-    a particular purpose are disclaimed. In no event shall the copyright
-    owner or contributors be liable for any direct, indirect, incidental,
-    special, exemplary, or consequential damages (including, but not
-    limited to, procurement of substitute goods or services; loss of use,
-    data, or profits; or business interruption) however caused and on any
-    theory of liability, whether in contract, strict liability, or tort
-    (including negligence or otherwise) arising in any way out of the use
-    of this software, even if advised of the possibility of such
-    damage.**
+Please refer to the ``LICENSE`` and ``NOTICE`` files for more detailed
+information.

{braindecode-0.8.1 → braindecode-1.1.0}/README.rst

@@ -110,37 +110,13 @@ as well as the `MNE-Python <https://mne.tools>`_ software that is used by braind
 Licensing
 ^^^^^^^^^
 
-Braindecode is **BSD-licenced** (BSD-3-Clause):
-
-    This software is OSI Certified Open Source Software.
-    OSI Certified is a certification mark of the Open Source Initiative.
-
-    Copyright (c) 2011-2022, authors of Braindecode.
-    All rights reserved.
-
-    Redistribution and use in source and binary forms, with or without
-    modification, are permitted provided that the following conditions are met:
-
-    * Redistributions of source code must retain the above copyright notice,
-      this list of conditions and the following disclaimer.
-
-    * Redistributions in binary form must reproduce the above copyright notice,
-      this list of conditions and the following disclaimer in the documentation
-      and/or other materials provided with the distribution.
-
-    * Neither the names of braindecode authors nor the names of any
-      contributors may be used to endorse or promote products derived from
-      this software without specific prior written permission.
-
-    **This software is provided by the copyright holders and contributors
-    "as is" and any express or implied warranties, including, but not
-    limited to, the implied warranties of merchantability and fitness for
-    a particular purpose are disclaimed. In no event shall the copyright
-    owner or contributors be liable for any direct, indirect, incidental,
-    special, exemplary, or consequential damages (including, but not
-    limited to, procurement of substitute goods or services; loss of use,
-    data, or profits; or business interruption) however caused and on any
-    theory of liability, whether in contract, strict liability, or tort
-    (including negligence or otherwise) arising in any way out of the use
-    of this software, even if advised of the possibility of such
-    damage.**
+This project is primarily licensed under the BSD-3-Clause License.
+
+Additional Components
+~~~~~~~~~~~~~~~~~~~~~
+
+Some components within this repository are licensed under the Creative Commons Attribution-NonCommercial 4.0 International
+License.
+
+Please refer to the ``LICENSE`` and ``NOTICE`` files for more detailed
+information.

{braindecode-0.8.1 → braindecode-1.1.0}/braindecode/__init__.py

@@ -1,7 +1,6 @@
-from .version import __version__
-
 from .classifier import EEGClassifier
 from .regressor import EEGRegressor
+from .version import __version__
 
 __all__ = [
     "__version__",

braindecode-1.1.0/braindecode/augmentation/__init__.py

@@ -0,0 +1,50 @@
+"""
+Utilities for data augmentation.
+"""
+
+from . import functional
+from .base import AugmentedDataLoader, Compose, IdentityTransform, Transform
+from .transforms import (
+    BandstopFilter,
+    ChannelsDropout,
+    ChannelsShuffle,
+    ChannelsSymmetry,
+    FrequencyShift,
+    FTSurrogate,
+    GaussianNoise,
+    MaskEncoding,
+    Mixup,
+    SegmentationReconstruction,
+    SensorsRotation,
+    SensorsXRotation,
+    SensorsYRotation,
+    SensorsZRotation,
+    SignFlip,
+    SmoothTimeMask,
+    TimeReverse,
+)
+
+__all__ = [
+    "Transform",
+    "IdentityTransform",
+    "Compose",
+    "AugmentedDataLoader",
+    "TimeReverse",
+    "SignFlip",
+    "FTSurrogate",
+    "ChannelsShuffle",
+    "ChannelsDropout",
+    "GaussianNoise",
+    "ChannelsSymmetry",
+    "SmoothTimeMask",
+    "BandstopFilter",
+    "FrequencyShift",
+    "SensorsRotation",
+    "SensorsZRotation",
+    "SensorsYRotation",
+    "SensorsXRotation",
+    "Mixup",
+    "SegmentationReconstruction",
+    "MaskEncoding",
+    "functional",
+]

{braindecode-0.8.1 → braindecode-1.1.0}/braindecode/augmentation/base.py

@@ -5,31 +5,28 @@
 #          Valentin Iovene <val@too.gy>
 # License: BSD (3-clause)
 
-from typing import List, Tuple, Any, Optional, Union, Callable
 from numbers import Real
+from typing import Any, Callable, Optional, Union
 
-from sklearn.utils import check_random_state
 import torch
+from sklearn.utils import check_random_state
 from torch import Tensor, nn
 from torch.utils.data import DataLoader
 from torch.utils.data._utils.collate import default_collate
 
 from .functional import identity
 
-Batch = List[Tuple[torch.Tensor, int, Any]]
+Batch = list[tuple[torch.Tensor, int, Any]]
 Output = Union[
     # just outputting X
     torch.Tensor,
     # outputting (X, y) where y can be a tensor or tuple of tensors
-    Tuple[torch.Tensor, Union[torch.Tensor, Tuple[torch.Tensor, ...]]]
+    tuple[torch.Tensor, Union[torch.Tensor, tuple[torch.Tensor, ...]]],
 ]
 # (X, y) -> (X', y') where y' can be a tensor or a tuple of tensors
 Operation = Callable[
     [torch.Tensor, torch.Tensor],
-    Tuple[
-        torch.Tensor,
-        Union[torch.Tensor, Tuple[torch.Tensor, ...]]
-    ]
+    tuple[torch.Tensor, Union[torch.Tensor, tuple[torch.Tensor, ...]]],
 ]
 
 
@@ -50,18 +47,20 @@ class Transform(torch.nn.Module):
         Used to decide whether or not to transform given the probability
         argument. Defaults to None.
     """
+
     operation: Operation
 
     def __init__(self, probability=1.0, random_state=None):
         super().__init__()
         if self.forward.__func__ is Transform.forward:
-            assert callable(self.operation), "operation should be a " \
-                                             "``callable``. "
+            assert callable(self.operation), "operation should be a ``callable``. "
 
         assert isinstance(probability, Real), (
-            f"probability should be a ``real``. Got {type(probability)}.")
-        assert probability <= 1. and probability >= 0., \
+            f"probability should be a ``real``. Got {type(probability)}."
+        )
+        assert probability <= 1.0 and probability >= 0.0, (
             "probability should be between 0 and 1."
+        )
         self._probability = probability
         self.rng = check_random_state(random_state)
 
@@ -108,8 +107,9 @@ class Transform(torch.nn.Module):
         if num_valid > 0:
             # Uses the mask to define the output
             out_X[mask, ...], tr_y = self.operation(
-                out_X[mask, ...], out_y[mask],
-                **self.get_augmentation_params(out_X[mask, ...], out_y[mask])
+                out_X[mask, ...],
+                out_y[mask],
+                **self.get_augmentation_params(out_X[mask, ...], out_y[mask]),
             )
             # Apply the operation defining the Transform to the whole batch
             if isinstance(tr_y, tuple):
@@ -125,11 +125,10 @@ class Transform(torch.nn.Module):
             return out_X
 
     def _get_mask(self, batch_size, device) -> torch.Tensor:
-        """Samples whether to apply operation or not over the whole batch
-        """
-        return torch.as_tensor(
-            self.probability > self.rng.uniform(size=batch_size)
-        ).to(device)
+        """Samples whether to apply operation or not over the whole batch"""
+        return torch.as_tensor(self.probability > self.rng.uniform(size=batch_size)).to(
+            device
+        )
 
     @property
     def probability(self):
@@ -141,7 +140,9 @@ class IdentityTransform(Transform):
 
     Transform that does not change the input.
     """
-    operation = staticmethod(identity)
+
+    operation = staticmethod(identity)  # type: ignore[assignment]
+    # https://github.com/python/mypy/issues/4574
 
 
 class Compose(Transform):
@@ -167,8 +168,8 @@ class Compose(Transform):
 
 
 def _make_collateable(transform, device=None):
-    """ Wraps a transform to make it collateable.
-        with device control. """
+    """Wraps a transform to make it collateable.
+    with device control."""
 
     def _collate_fn(batch):
         collated_batch = default_collate(batch)
@@ -205,7 +206,7 @@ class AugmentedDataLoader(DataLoader):
                 "to pass transform"
             )
         if transforms is None or (
-                isinstance(transforms, list) and len(transforms) == 0
+            isinstance(transforms, list) and len(transforms) == 0
         ):
             self.collated_tr = _make_collateable(IdentityTransform(), device=device)
         elif isinstance(transforms, (Transform, nn.Module)):
@@ -218,8 +219,4 @@ class AugmentedDataLoader(DataLoader):
                 "or a list of Transform objects."
             )
 
-        super().__init__(
-            dataset,
-            collate_fn=self.collated_tr,
-            **kwargs
-        )
+        super().__init__(dataset, collate_fn=self.collated_tr, **kwargs)