braindecode 1.8.0.dev183527593__tar.gz → 1.8.1.dev167868425__tar.gz

braindecode-1.8.1.dev167868425/NOTICE.txt

@@ -0,0 +1,78 @@
+# BRAINDECODE Notice
+
+## Licensed Components
+
+### BSD-3-Clause Licensed Files
+
+All files within the `braindecode/` package are licensed under the BSD-3-Clause
+License, except for those listed in the sections below.
+
+### 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`
+- `braindecode/models/meta_neuromotor.py`
+- `braindecode/models/brainmodule.py`
+
+As well as class later imported into the `braindecode.models.module` named as GeneralizedGaussianFilter.
+
+The `meta_neuromotor.py` file is a derivative of
+`facebookresearch/generic-neuromotor-interface`, released by Meta Platforms,
+Inc. under CC BY-NC 4.0, and inherits the same noncommercial terms.
+
+### CC BY-NC-SA 4.0 Licensed Files
+
+The following components are licensed under the Creative Commons
+Attribution-NonCommercial-ShareAlike 4.0 International License:
+
+- `braindecode/models/emg2qwerty.py`
+- `braindecode/models/vemg2pose.py`
+
+The `emg2qwerty.py` file is a derivative of `facebookresearch/emg2qwerty`,
+released by Meta Platforms, Inc. under CC BY-NC-SA 4.0, and inherits the
+same noncommercial ShareAlike terms.
+
+### MIT Licensed Files
+
+The following components are licensed under the MIT License:
+
+- `braindecode/models/ctnet.py`
+- `braindecode/models/dance.py`
+- `braindecode/models/medformer.py`
+- `braindecode/models/tcformer.py`
+- `braindecode/models/ifnet.py`
+- `braindecode/models/mirepnet.py`
+
+### Apache-2.0 Licensed Files
+
+The following components are licensed under the Apache License 2.0:
+
+- `braindecode/models/mvpformer.py`
+- `braindecode/models/zuna.py`
+- `braindecode/models/luna.py`
+- `braindecode/models/brant.py`
+- `braindecode/models/diver1.py`
+- `braindecode/models/mscformer.py`
+
+### USC Academic License Files
+
+[License](https://github.com/ShanechiLab/BaRISTA/blob/master/LICENSE.md)
+
+- `braindecode/models/barista.py`
+- `docs/_static/model/barista.jpeg`
+
+## 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/)
+- [CC BY-NC-SA 4.0 License](https://creativecommons.org/licenses/by-nc-sa/4.0/)
+- [MIT License](https://opensource.org/licenses/MIT)
+- [Apache-2.0 License](https://www.apache.org/licenses/LICENSE-2.0)
+- [USC Academic License](https://github.com/ShanechiLab/BaRISTA/blob/master/LICENSE.md)
+
+## Note
+
+This list covers files whose own headers declare a license other than
+BSD-3-Clause. Per-file provenance review of the remaining models is
+tracked separately.

{braindecode-1.8.0.dev183527593/braindecode.egg-info → braindecode-1.8.1.dev167868425}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.8.0.dev183527593
+Version: 1.8.1.dev167868425
 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>
@@ -22,8 +22,8 @@ Requires-Python: >=3.11
 Description-Content-Type: text/x-rst
 License-File: LICENSE.txt
 License-File: NOTICE.txt
-Requires-Dist: torch>=2.0
-Requires-Dist: torchaudio>=2.0
+Requires-Dist: torch>=2.4
+Requires-Dist: torchaudio>=2.4
 Requires-Dist: mne>=1.11.0
 Requires-Dist: mne_bids>=0.18
 Requires-Dist: h5py
@@ -34,7 +34,6 @@ Requires-Dist: pandas
 Requires-Dist: wfdb>=4.3.1
 Requires-Dist: linear_attention_transformer
 Requires-Dist: docstring_inheritance
-Requires-Dist: rotary_embedding_torch
 Requires-Dist: pydantic>=2.0
 Provides-Extra: moabb
 Requires-Dist: moabb>=1.4.3; extra == "moabb"
@@ -45,19 +44,23 @@ Requires-Dist: huggingface_hub[torch]>=0.20.0; extra == "hub"
 Requires-Dist: zarr>=3.0; extra == "hub"
 Provides-Extra: viz
 Requires-Dist: captum>=0.7.0; extra == "viz"
+Provides-Extra: sae
+Requires-Dist: sae-lens==6.51.3; extra == "sae"
 Provides-Extra: tests
 Requires-Dist: pytest<9.1; extra == "tests"
 Requires-Dist: pytest-cov; extra == "tests"
+Requires-Dist: pytest-xdist>=3.8.0; extra == "tests"
 Requires-Dist: codecov; extra == "tests"
 Requires-Dist: pytest_cases; extra == "tests"
 Requires-Dist: mypy; extra == "tests"
+Requires-Dist: ipython; extra == "tests"
 Requires-Dist: transformers>=4.57.0; extra == "tests"
 Requires-Dist: bids_validator; extra == "tests"
 Provides-Extra: typing
 Requires-Dist: exca>=0.5.10; extra == "typing"
 Requires-Dist: numpydantic>=1.7; extra == "typing"
 Provides-Extra: docs
-Requires-Dist: sphinx_gallery; extra == "docs"
+Requires-Dist: sphinx_gallery>=0.17; extra == "docs"
 Requires-Dist: sphinx_rtd_theme; extra == "docs"
 Requires-Dist: sphinx-autodoc-typehints; extra == "docs"
 Requires-Dist: sphinx-autobuild; extra == "docs"
@@ -260,7 +263,10 @@ 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.
+Some components within this repository are licensed under other licenses, including
+Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0), Creative
+Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0), MIT
+and Apache-2.0.
 
-Please refer to the ``LICENSE`` and ``NOTICE`` files for more detailed information.
+Please refer to the ``LICENSE`` and ``NOTICE`` files for the per-file list and more
+detailed information.

{braindecode-1.8.0.dev183527593 → braindecode-1.8.1.dev167868425}/README.rst

@@ -170,7 +170,10 @@ 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.
+Some components within this repository are licensed under other licenses, including
+Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0), Creative
+Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0), MIT
+and Apache-2.0.
 
-Please refer to the ``LICENSE`` and ``NOTICE`` files for more detailed information.
+Please refer to the ``LICENSE`` and ``NOTICE`` files for the per-file list and more
+detailed information.

{braindecode-1.8.0.dev183527593 → braindecode-1.8.1.dev167868425}/braindecode/augmentation/functional.py

@@ -1002,7 +1002,7 @@ def sensors_rotation(
         ``mne`` through::
 
          >>> ten_twenty_montage = mne.channels.make_standard_montage(
-         ...    'standard_1020'
+         ...    'standard_1020'  # 'colin27_1020' on MNE >= 1.13
          ... ).get_positions()['ch_pos']
     axis : 'x' | 'y' | 'z'
         Axis around which to rotate.

{braindecode-1.8.0.dev183527593 → braindecode-1.8.1.dev167868425}/braindecode/augmentation/transforms.py

@@ -13,6 +13,8 @@ import numpy as np
 import torch
 from mne.channels import make_standard_montage
 
+from braindecode.util import resolve_montage_name
+
 from .base import Transform
 from .functional import (
     amplitude_scale,
@@ -703,7 +705,7 @@ def _get_standard_10_20_positions(raw_or_epoch=None, ordered_ch_names=None):
     )
     if ordered_ch_names is None:
         ordered_ch_names = raw_or_epoch.info["ch_names"]
-    ten_twenty_montage = make_standard_montage("standard_1020")
+    ten_twenty_montage = make_standard_montage(resolve_montage_name("standard_1020"))
     positions_dict = ten_twenty_montage.get_positions()["ch_pos"]
     positions_subdict = {
         k: positions_dict[k] for k in ordered_ch_names if k in positions_dict
@@ -729,7 +731,7 @@ class SensorsRotation(Transform):
         `mne` through::
 
          >>> ten_twenty_montage = mne.channels.make_standard_montage(
-         ...    'standard_1020'
+         ...    'standard_1020'  # 'colin27_1020' on MNE >= 1.13
          ... ).get_positions()['ch_pos']
 
     axis : 'x' | 'y' | 'z', optional

braindecode-1.8.1.dev167868425/braindecode/datasets/_notebook_viewer.py

@@ -0,0 +1,255 @@
+# Authors: Bruno Aristimunha <b.aristimunha@gmail.com>
+#
+# License: BSD (3-clause)
+"""Serverless in-notebook viewer for file-backed recordings.
+
+The bytes on disk behind a dataset element are inlined in the cell output
+as base64 and handed to the deployed eegdash-viewer over its ``postMessage``
+bridge (``docs/embedding.md`` in https://github.com/eegdash/eegdash-viewer):
+no server, no CORS. The output is an iframe plus an inline script, so it
+renders when the cell ran in your session or the saved notebook is trusted
+(``jupyter trust``).
+"""
+
+from __future__ import annotations
+
+import base64
+import json
+import os
+import uuid
+from pathlib import Path
+from urllib.parse import urlsplit
+
+import mne_bids
+from mne.utils import _soft_import
+
+CDN = "https://eegdash.github.io/eegdash-viewer"
+MAX_BYTES = 64 * 2**20  # base64 output per call; it is saved with the notebook
+EXTENSIONS = {
+    ".set",
+    ".edf",
+    ".bdf",
+    ".vhdr",
+    ".fif",
+    ".snirf",
+    ".nwb",
+}  # viewer readers
+_BIDS = {"eeg", "ieeg", "emg", "meg", "nirs"}
+_SIBLINGS = {".vhdr": (".eeg", ".vmrk"), ".set": (".fdt",)}  # travel with the header
+_HEADER = {s: h for h, ss in _SIBLINGS.items() for s in ss}  # data file -> header
+
+_SCRIPT = """<iframe id=%(id)s title="eegdash trace viewer" style="width:100%%;height:%(height)spx;
+border:1px solid var(--jp-border-color1,#d9dce1);border-radius:6px;background:transparent"></iframe>
+<script>
+(function () {
+  var self = document.currentScript, id = %(id)s;   // Lab re-runs scripts in place; VS Code/nbclassic elsewhere
+  var frame = (self && self.previousElementSibling && self.previousElementSibling.tagName === "IFRAME")
+    ? self.previousElementSibling : document.getElementById(id);
+  if (!frame) { console.error("eegdash viewer: output iframe " + id + " not found"); return; }
+  var payload = %(payload)s, origin = %(origin)s, files = null, pose = null;
+  function decode(b64) {
+    if (Uint8Array.fromBase64) return Uint8Array.fromBase64(b64);
+    var bin = atob(b64), out = new Uint8Array(bin.length);
+    for (var i = 0; i < bin.length; i++) out[i] = bin.charCodeAt(i);
+    return out;
+  }
+  function send(target) {
+    try {
+      if (!files) {
+        files = payload.files.map(function (f) { return new File([decode(f.b64)], f.name); });
+        pose = payload.pose ? "data:application/json;base64," + payload.pose : null;
+        payload = null;
+      }
+      frame.contentWindow.postMessage({ type: "eegdash-viewer:open", files: files, pose: pose }, target || origin);
+    } catch (err) {
+      frame.insertAdjacentHTML("afterend", '<div style="font:12px system-ui;color:#b3261e">eegdash viewer: '
+        + String(err.message).replace(/</g, "&lt;") + "</div>");
+    }
+  }
+  window.addEventListener("message", function onMessage(e) {
+    if (e.source === frame.contentWindow && e.data && e.data.type === "eegdash-viewer:ready") send(e.origin);
+    else if (!frame.isConnected) window.removeEventListener("message", onMessage);
+  });
+  frame.src = %(src)s;   // after the listener, so "ready" can never precede it
+})();
+</script>"""
+
+
+def recording_files(recording: Path) -> tuple[list[Path], Path | None]:
+    """``(files, pose)`` to inline for a recording: the header first, then the
+    split-format siblings, the BIDS-inherited ``_channels.tsv``/``_events.tsv``
+    (``mne_bids`` parses the name; plain names get none) and, separately, the
+    ``<prefix>_desc-pose.json`` hand-pose sidecar next to it. Symlinks keep
+    their name (git-annex/datalad)."""
+    rec = Path(recording)
+    if not rec.exists():
+        raise ValueError(
+            f"{rec.name}: file not found (a datalad symlink may need `datalad get`)"
+        )
+    if (
+        rec.is_dir()
+        or rec.suffix.lower() not in EXTENSIONS
+        or rec.stem.endswith("_epo")
+    ):
+        raise ValueError(
+            f"{rec.name}: the viewer opens raw recordings in {' '.join(sorted(EXTENSIONS))}"
+        )
+    sidecars: list[Path | None] = []
+    try:
+        bids = mne_bids.get_bids_path_from_fname(rec, check=False)
+        if bids.subject is not None:  # hyphen-free names parse, but are not BIDS
+            sidecars = [
+                bids.find_matching_sidecar(
+                    suffix=s, extension=".tsv", on_error="ignore"
+                )
+                for s in ("channels", "events")
+            ]
+    except (
+        KeyError,
+        ValueError,
+    ):  # not a BIDS name / unknown entity somewhere in the tree
+        pass
+    files = [rec]
+    for p in [
+        rec.with_suffix(e) for e in _SIBLINGS.get(rec.suffix.lower(), ())
+    ] + sidecars:
+        if p and p.is_file():
+            if p not in files:
+                files.append(p)
+        elif p and p.is_symlink():  # dangling: git-annex/datalad content not fetched
+            raise ValueError(f"{p.name}: dangling symlink (try `datalad get`)")
+    stem, _, token = rec.stem.rpartition("_")
+    pose = rec.with_name((stem if token in _BIDS else rec.stem) + "_desc-pose.json")
+    return files, pose if pose.is_file() else None
+
+
+def build_viewer_html(
+    recording: Path,
+    *,
+    height: int = 520,
+    cdn_url: str = CDN,
+    max_bytes: int = MAX_BYTES,
+) -> str:
+    """Viewer iframe + inlined bytes + bridge script for one recording."""
+    url = urlsplit(cdn_url)
+    if (
+        url.scheme not in ("http", "https")
+        or not url.netloc
+        or url.username
+        or url.query
+        or url.fragment
+        or url.path.endswith(("index.html", "index.htm"))
+    ):
+        raise ValueError(
+            f"cdn_url must be the viewer's base http(s) URL, got {cdn_url!r}"
+        )
+    files, pose = recording_files(recording)
+    encoded = sum(
+        4 * -(-p.stat().st_size // 3) for p in files + ([pose] if pose else [])
+    )
+    if encoded > max_bytes:
+        raise ValueError(
+            f"{files[0].name}: {encoded / 2**20:.1f} MiB of base64 would be inlined into the "
+            f"notebook output (max_bytes={max_bytes / 2**20:.1f} MiB); crop/downsample or raise it"
+        )
+    rec = files[0]
+    # The viewer picks the recording by its *_<datatype>.<ext> name: plain names
+    # are posted as <stem>_eeg<ext>, an EEGLAB .fdt next to the posted .set name.
+    head = (
+        rec.name
+        if rec.stem.rpartition("_")[2] in _BIDS
+        else f"{rec.stem}_eeg{rec.suffix.lower()}"
+    )
+    names = [head] + [
+        head.rsplit("_", 1)[0] + "_eeg.fdt" if p.suffix.lower() == ".fdt" else p.name
+        for p in files[1:]
+    ]
+    b64 = [base64.b64encode(p.read_bytes()).decode() for p in files]
+    literals = {
+        "id": f"eegdash-viewer-{uuid.uuid4().hex[:8]}",
+        "height": int(height),
+        "payload": {
+            "files": [{"name": n, "b64": b} for n, b in zip(names, b64)],
+            "pose": base64.b64encode(pose.read_bytes()).decode() if pose else None,
+        },
+        "origin": f"{url.scheme}://{url.netloc}",
+        "src": f"{url.geturl().rstrip('/')}/index.html?embed=1",
+    }
+    return _SCRIPT % {
+        k: json.dumps(v).replace("<", "\\u003c") for k, v in literals.items()
+    }
+
+
+def _recording(dataset, index: int) -> Path:
+    """File behind ``dataset.datasets[index]``: the one its mne ``raw`` reads (a
+    data file maps back to its header; lazily downloading datasets fetch it when
+    ``raw`` is accessed) or the recorded ``description["path"]``."""
+    ds = dataset.datasets[index]
+    names = [
+        Path(f)
+        for f in getattr(getattr(ds, "raw", None), "filenames", None) or ()
+        if isinstance(f, (str, os.PathLike))
+    ]
+    if len(names) > 1:
+        raise ValueError(
+            f"{type(dataset).__name__}[{index}]: split recordings are not supported"
+        )
+    desc = getattr(ds, "description", None)  # dict or pandas Series
+    recorded = desc.get("path") if desc is not None else None
+    path = (
+        names[0]
+        if names
+        else Path(recorded)
+        if isinstance(recorded, (str, os.PathLike))
+        else None
+    )
+    if path is None:
+        raise ValueError(
+            f"{type(dataset).__name__}[{index}] is not backed by a recording file"
+        )
+    header = path.with_suffix(_HEADER.get(path.suffix.lower(), path.suffix))
+    return header if header.is_file() else path
+
+
+def plot(
+    dataset,
+    index: int = 0,
+    *,
+    height: int = 520,
+    cdn_url: str = CDN,
+    max_bytes: int = MAX_BYTES,
+):
+    """Show one recording in the eegdash-viewer inside a Jupyter cell.
+
+    Serverless: the recording bytes (as on disk) are inlined in the output and
+    pushed to the viewer at ``cdn_url`` over ``postMessage``; the output renders
+    when the cell ran in your session or the notebook is trusted. A
+    ``*_desc-pose.json`` sidecar next to the recording adds the synchronized
+    hand-pose panel. Needs IPython (soft dependency).
+
+    Parameters
+    ----------
+    index : int
+        Recording to display.
+    height : int
+        Viewer height in pixels.
+    cdn_url : str
+        Base URL of a deployed eegdash-viewer.
+    max_bytes : int
+        Refuse to inline more than this much base64 (default 64 MiB); the
+        payload is saved with the notebook and, like any cell output, stays
+        referenced by IPython's ``Out`` history for the session.
+
+    Returns
+    -------
+    IPython.display.HTML
+    """
+    ipython = _soft_import("IPython", purpose=f"{type(dataset).__name__}.plot()")
+    return ipython.display.HTML(
+        build_viewer_html(
+            _recording(dataset, index),
+            height=height,
+            cdn_url=cdn_url,
+            max_bytes=max_bytes,
+        )
+    )

{braindecode-1.8.0.dev183527593 → braindecode-1.8.1.dev167868425}/braindecode/datasets/base.py

@@ -5,6 +5,7 @@
 #          Simon Brandt <simonbrandt@protonmail.com>
 #          David Sabbagh <dav.sabbagh@gmail.com>
 #          Robin Schirrmeister <robintibor@gmail.com>
+#          Sarthak Tayal <sarthaktayal2@gmail.com>
 #
 # License: BSD (3-clause)
 
@@ -32,6 +33,7 @@ from mne.utils.docs import deprecated
 from torch.utils.data import ConcatDataset, Dataset, IterableDataset
 from typing_extensions import TypeVar
 
+from ._notebook_viewer import plot as _viewer_plot
 from .bids.hub import HubDatasetMixin
 from .bids.hub_io import _restore_nan_from_json
 from .registry import register_dataset
@@ -59,6 +61,7 @@ def _html_row(label, value):
 
 _METADATA_INTERNAL_COLS = {
     "i_window_in_trial",
+    "i_trial_in_dataset",
     "i_start_in_trial",
     "i_stop_in_trial",
     "target",
@@ -1173,6 +1176,8 @@ class BaseConcatDataset(ConcatDataset, HubDatasetMixin, Generic[T]):
         If True, defer computing cumulative sizes until length or item access.
     """
 
+    plot = _viewer_plot  # eegdash-viewer embed; a direct member so the API docs list it
+
     datasets: list[T]
 
     def __init__(
@@ -1374,12 +1379,29 @@ class BaseConcatDataset(ConcatDataset, HubDatasetMixin, Generic[T]):
                 "datasets are WindowsDataset."
             )
 
+        for ds in self.datasets:
+            if hasattr(ds, "_windows") and ds._windows is not None:
+                df = ds._windows.metadata
+            else:
+                df = ds.metadata
+            if (
+                "i_trial_in_dataset" in df.columns
+                and "i_trial_in_dataset" in ds.description
+            ):
+                raise ValueError(
+                    "Dataset descriptions cannot contain the reserved window "
+                    "metadata key 'i_trial_in_dataset'."
+                )
+
         all_dfs = list()
         for ds in self.datasets:
             if hasattr(ds, "_windows") and ds._windows is not None:
                 df = ds._windows.metadata
             else:
                 df = ds.metadata
+            if isinstance(df, pd.DataFrame):
+                # copy so the description columns stay out of the dataset
+                df = df.copy()
             for k, v in ds.description.items():
                 df[k] = v
             all_dfs.append(df)

{braindecode-1.8.0.dev183527593 → braindecode-1.8.1.dev167868425}/braindecode/datasets/tuh.py

@@ -25,6 +25,7 @@ import pandas as pd
 from joblib import Parallel, delayed
 
 from braindecode.datasets.base import BaseConcatDataset, RawDataset
+from braindecode.util import resolve_montage_name
 
 
 class TUH(BaseConcatDataset):
@@ -182,7 +183,9 @@ class TUH(BaseConcatDataset):
         }
         raw.rename_channels(mapping_strip)
 
-        montage1005 = mne.channels.make_standard_montage("standard_1005")
+        montage1005 = mne.channels.make_standard_montage(
+            resolve_montage_name("standard_1005")
+        )
         mapping_eeg_names = {
             c.upper(): c for c in montage1005.ch_names if c.upper() in raw.ch_names
         }
@@ -207,7 +210,9 @@ class TUH(BaseConcatDataset):
 
     @staticmethod
     def _set_montage(raw):
-        montage = mne.channels.make_standard_montage("standard_1005")
+        montage = mne.channels.make_standard_montage(
+            resolve_montage_name("standard_1005")
+        )
         raw.set_montage(montage, on_missing="ignore")
 
     @staticmethod

{braindecode-1.8.0.dev183527593 → braindecode-1.8.1.dev167868425}/braindecode/datautil/serialization.py

@@ -1,6 +1,7 @@
 """Convenience functions for storing and loading of windows datasets."""
 
 # Authors: Lukas Gemein <l.gemein@gmail.com>
+#          Sarthak Tayal <sarthaktayal2@gmail.com>
 #
 # License: BSD (3-clause)
 
@@ -270,7 +271,9 @@ def _load_parallel(path, i, preload, is_raw, has_stored_windows):
     else:
         window_kwargs = _load_kwargs_json("window_kwargs", sub_dir)
         windows_ds_kwargs = [
-            kwargs[1] for kwargs in window_kwargs if kwargs[0] == "WindowsDataset"
+            kwargs[1]
+            for kwargs in window_kwargs
+            if kwargs[0] in ("WindowsDataset", "EEGWindowsDataset")
         ]
         windows_ds_kwargs = windows_ds_kwargs[0] if len(windows_ds_kwargs) == 1 else {}
         if is_raw:

{braindecode-1.8.0.dev183527593 → braindecode-1.8.1.dev167868425}/braindecode/functional/__init__.py

@@ -8,6 +8,7 @@ from .functions import (
     iou_1d,
     pairwise_iou_1d,
     plv_time,
+    rotate_pairs,
     safe_log,
     sinusoidal_positional_encoding,
     square,
@@ -25,6 +26,7 @@ __all__ = [
     "iou_1d",
     "pairwise_iou_1d",
     "plv_time",
+    "rotate_pairs",
     "safe_log",
     "sinusoidal_positional_encoding",
     "square",