Braindecode 1.5.0.dev982__py3-none-any.whl → 1.5.0.dev984__py3-none-any.whl

braindecode/datasets/base.py

@@ -298,34 +298,95 @@ def _build_windowed_repr(
 
 
 def _zarr_to_memmap(zarr_path, group_name):
-    """Decompress a zarr array to a float64 ``.npy`` memmap (one-time, atomic).
-
-    Saved as float64 so MNE objects can wrap the memmap zero-copy
-    (MNE requires float64 internally).  Copy-on-write mode (``'c'``)
-    means preprocessing writes go to RAM, the file stays untouched.
+    """Materialise a zarr array as a float64 ``.npy`` memmap, race-free.
+
+    Decompress the ``group_name`` array inside ``zarr_path`` into a
+    float64 ``.npy`` file inside a sibling ``.<zarr>_memmap/`` cache
+    directory and return the path to that file.  The dtype is float64
+    because MNE objects wrap the memmap zero-copy and MNE requires
+    float64 internally.  Callers open the result with
+    ``np.load(path, mmap_mode='c')`` so preprocessing writes land in
+    anonymous copy-on-write pages and never mutate the on-disk file.
+
+    Concurrency contract
+    --------------------
+    Safe to call from any number of threads or processes concurrently
+    on the same ``(zarr_path, group_name)``, on local POSIX, NFSv3,
+    Lustre and SMB.  The published ``.npy`` file is **created exactly
+    once and never replaced**: subsequent callers either hit the
+    already-materialised file on the fast path or lose the
+    publication race and discard their own copy.
+
+    The atomic publication step is ``os.link``, not ``rename``.
+    ``os.link(tmp, dst)`` either creates ``dst`` pointing at ``tmp``'s
+    inode or fails with :class:`FileExistsError` when ``dst`` already
+    exists — the "create only if absent" primitive we need, available
+    on every POSIX filesystem.  ``rename`` was the wrong primitive:
+    it atomically *replaces* its destination, which unlinks the inode
+    that concurrent readers may already have ``mmap``'d.  On NFSv3
+    that produced ``.nfsXXXX`` silly-rename files and intermittent
+    ``SIGBUS`` when workers page-faulted on the unlinked inode.
+
+    Under ``N`` concurrent writers, every racer materialises the data
+    into its own per-pid temp file and then tries to publish it; at
+    most one ``os.link`` wins, and the losers silently ``unlink``
+    their temp file in a ``finally`` block.  The wasted I/O scales
+    with the number of racers; for the workloads braindecode targets
+    (a handful of concurrent workers per group) it is negligible.
+
+    Crash recovery: if a writer dies mid-run the only debris is a
+    ``<group>.<pid>.tmp.npy`` file.  Such files are namespaced by pid
+    so they can never collide with a live writer, they are never
+    read by the fast path, and they do not prevent subsequent
+    callers from producing a correct result.  Caches written by
+    earlier braindecode versions remain valid: the fast path only
+    requires ``<group>.npy`` to exist.
     """
     zarr_p = Path(zarr_path)
     cache_dir = zarr_p.parent / f".{zarr_p.name}_memmap"
     npy_path = cache_dir / f"{group_name}.npy"
 
-    if not npy_path.exists():
-        arr = zarr.open(zarr_path, mode="r")[group_name]["data"]
-        cache_dir.mkdir(parents=True, exist_ok=True)
-        tmp_path = cache_dir / f"{group_name}.{os.getpid()}.tmp.npy"
-        # Write chunk-by-chunk to avoid materializing the full array in RAM
+    # Fast path: the file is already fully materialised.  A single
+    # stat() is all the steady-state workload pays.
+    if npy_path.exists():
+        return npy_path
+
+    cache_dir.mkdir(parents=True, exist_ok=True)
+    arr = zarr.open(zarr_path, mode="r")[group_name]["data"]
+    tmp_path = cache_dir / f"{group_name}.{os.getpid()}.tmp.npy"
+    try:
+        # Write chunk-by-chunk to avoid materialising the whole zarr
+        # array in RAM.
         mm = np.lib.format.open_memmap(
             tmp_path, mode="w+", dtype=np.float64, shape=arr.shape
         )
-        chunk0 = arr.chunks[0] if hasattr(arr, "chunks") else arr.shape[0]
-        for start in range(0, arr.shape[0], chunk0):
-            end = min(start + chunk0, arr.shape[0])
-            mm[start:end] = np.asarray(arr[start:end], dtype=np.float64)
-        mm.flush()
-        del mm
         try:
-            tmp_path.rename(npy_path)
-        except OSError:
-            tmp_path.unlink(missing_ok=True)
+            chunk0 = arr.chunks[0] if hasattr(arr, "chunks") else arr.shape[0]
+            for start in range(0, arr.shape[0], chunk0):
+                end = min(start + chunk0, arr.shape[0])
+                mm[start:end] = np.asarray(arr[start:end], dtype=np.float64)
+            mm.flush()
+        finally:
+            del mm
+
+        # Atomic publication.  ``os.link`` never replaces an existing
+        # destination, so the published inode is immutable for the
+        # lifetime of the cache directory.  Concurrent readers never
+        # see their mmap'd inode vanish from under them.
+        try:
+            os.link(tmp_path, npy_path)
+        except FileExistsError:
+            # Another writer beat us to it.  Their file is the
+            # canonical one; ours is discarded in the finally block.
+            pass
+    finally:
+        # Always remove our private per-pid temp file.  It is
+        # namespaced by pid so this cannot clobber a concurrent
+        # writer's in-progress tmp file.
+        try:
+            os.unlink(tmp_path)
+        except FileNotFoundError:
+            pass
 
     return npy_path
 

braindecode/models/attentionbasenet.py

@@ -1,5 +1,8 @@
 from __future__ import annotations
 
+# Authors: Sarthak Tayal <sarthaktayal2@gmail.com>
+#
+# License: BSD (3-clause)
 import math
 
 from einops.layers.torch import Rearrange
@@ -275,8 +278,6 @@ class AttentionBaseNet(EEGModuleMixin, nn.Module):
         activation: type[nn.Module] = nn.ELU,
         extra_params: bool = False,
     ):
-        super(AttentionBaseNet, self).__init__()
-
         super().__init__(
             n_outputs=n_outputs,
             n_chans=n_chans,

braindecode/models/syncnet.py

@@ -1,3 +1,7 @@
+# Authors: Sarthak Tayal <sarthaktayal2@gmail.com>
+#
+# License: BSD (3-clause)
+
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -51,11 +55,11 @@ class SyncNet(EEGModuleMixin, nn.Module):
         The initialization range for omega parameters using uniform
         distribution. Default is (0, 1).
     beta_init_values : tuple of float, optional
-        The initialization range for beta parameters using uniform
-        distribution. Default is (0, 1). Default is (0, 0.05).
+        The initialization range for beta (decay) parameters using uniform
+        distribution. Default is (0, 0.05).
     phase_init_values : tuple of float, optional
-        The initialization range for phase parameters using `normal`
-        distribution. Default is (0, 1). Default is (0, 0.05).
+        The initialization mean and standard deviation for phase
+        parameters using normal distribution. Default is (0, 0.05).
 
 
     Notes
@@ -146,12 +150,12 @@ class SyncNet(EEGModuleMixin, nn.Module):
         # Phase Shift
         self.phi_ini = nn.Parameter(
             torch.FloatTensor(1, 1, self.n_chans, self.num_filters).normal_(
-                self.beta_init_values[0], self.beta_init_values[1]
+                self.phase_init_values[0], self.phase_init_values[1]
             )
         )
         self.beta = nn.Parameter(
             torch.FloatTensor(1, 1, 1, self.num_filters).uniform_(
-                self.phase_init_values[0], self.phase_init_values[1]
+                self.beta_init_values[0], self.beta_init_values[1]
             )
         )
 
@@ -185,21 +189,21 @@ class SyncNet(EEGModuleMixin, nn.Module):
         # Output: (batch_size, n_chans, 1, n_times)
 
         # Compute the oscillatory component
+        # Shape: (1, filter_width, n_chans, num_filters)
         W_osc = self.amplitude * torch.cos(self.t * self.omega + self.phi_ini)
-        # W_osc is (1, filter_width, n_chans, 1)
 
         # Compute the decay component
-        t_squared = torch.pow(self.t, 2)  # Shape: (filter_width,)
-        t_squared_beta = t_squared * self.beta  # Shape: (filter_width, num_filters)
+        # Shape: (1, filter_width, 1, num_filters)
+        t_squared = torch.pow(self.t, 2)
+        t_squared_beta = t_squared * self.beta
         W_decay = torch.exp(-t_squared_beta)
-        # W_osc is (1, filter_width, 1, 1)
 
         # Combine oscillatory and decay components
-        # W shape: (1, n_chans, num_filters, filter_width)
+        # Shape: (1, filter_width, n_chans, num_filters)
         W = W_osc * W_decay
-        # W shape will be: (1, filter_width, n_chans, 1)
 
-        W = W.view(self.num_filters, self.n_chans, 1, self.filter_width)
+        # Permute to conv2d weight shape (out_channels, in_channels, kH, kW)
+        W = W.permute(3, 2, 0, 1).contiguous()
 
         # Apply convolution
         x_padded = self.pad_input(x.float())

braindecode/models/tsinception.py

@@ -1,4 +1,4 @@
-# Authors: Bruno Aristimunha <b.aristimunha>
+# Authors: Bruno Aristimunha <b.aristimunha@gmail.com>
 #
 # License: BSD (3-clause)
 

braindecode/version.py

@@ -1 +1 @@
-__version__ = "1.5.0.dev982"
+__version__ = "1.5.0.dev984"

{braindecode-1.5.0.dev982.dist-info → braindecode-1.5.0.dev984.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.5.0.dev982
+Version: 1.5.0.dev984
 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.5.0.dev982.dist-info → braindecode-1.5.0.dev984.dist-info}/RECORD

@@ -3,13 +3,13 @@ braindecode/classifier.py,sha256=7kC_oY_UzHEes_WWdCvEpiA1ZKxMeuLL5tIPp5rfcpg,962
 braindecode/eegneuralnet.py,sha256=xjE6aPZdCQPs29NIpy_m1GLMMC2WZ3Db0Fuh1-xE1h4,13827
 braindecode/regressor.py,sha256=KiMJpqCUPWA2k2JWk9HGYTzeoBqJ4gAKEudeUVcFZY4,9266
 braindecode/util.py,sha256=f8bNIwt-SwsHqheH_BADQxTtA9oPt3Lb7GFnoI-Huwc,14101
-braindecode/version.py,sha256=NViteJgXY2XCTtVnQLPK1Ak2pdsVODMJMzeRYTw6fbc,29
+braindecode/version.py,sha256=5WqXRAeL_e0Aq78aGQsveuaWwGcV_OgtE7s9XQXznkc,29
 braindecode/augmentation/__init__.py,sha256=4xune2QUK6KHMKsAqijF7I9eeiVbP0wEoQJjCNLNcKM,1081
 braindecode/augmentation/base.py,sha256=OJ1shOljI1yTY9zh2qWxQwivlY43sfx9Q-MAyMhxtPs,7338
 braindecode/augmentation/functional.py,sha256=q2k6mAXrujYlOZUndcjZN8e8b-6oJF1gGsORAI23hyE,43998
 braindecode/augmentation/transforms.py,sha256=x-3pwX0PtMHfSnPLGKNXbpTSk7j17Ci2FG_-646scg4,47268
 braindecode/datasets/__init__.py,sha256=rVOBadwqYBiMz5kl7nGiBOmMgr11xvjS4nuzzZTOn1U,1102
-braindecode/datasets/base.py,sha256=ghLcjCJqvUkJhtgHuZ5bHKIhDEUmaQrt-U2jw8wQ7T8,61697
+braindecode/datasets/base.py,sha256=7qFEJNOAe_05evfqxx-yrlUfG0eWaCuKkfOlj32ShwU,64701
 braindecode/datasets/bbci.py,sha256=SCm7OnCObotILQ0B1EdmZPoyJtzsRXpeU_gNKtqQLSc,19288
 braindecode/datasets/bcicomp.py,sha256=YWIsRYFvwBFHnd9CRxo_BBQWFg-rg0UirRTsd0Ml2Oc,7550
 braindecode/datasets/chb_mit.py,sha256=XGUVtADLHLTa5Ldanyn8msAJY0SQh-eV9keHnWO6n3A,7231
@@ -41,7 +41,7 @@ braindecode/functional/functions.py,sha256=G29BPqIndSdvJiilDBZovklK_einjzFDyfn-3
 braindecode/functional/initialization.py,sha256=Nn773hEMWVYDYeunFv73tdxdtcYwXFC7M4HzrfWsaHQ,1385
 braindecode/models/__init__.py,sha256=sLJIAT_N2ZblDjt9sbOLQvfdTLWWmxgV8VzfS0nIqYs,3071
 braindecode/models/atcnet.py,sha256=JzLV92WYWlhQol8AqE9xeWvlnWN_uswYhVOkMnuBCXo,32231
-braindecode/models/attentionbasenet.py,sha256=nIvKen5FTuGBrdkodpIxfNekYuoTg-DpjXjlbrR06ZQ,30454
+braindecode/models/attentionbasenet.py,sha256=k7ar7aEjANudPu7krAZsRx-ag61ugirS1Xn7qFKNfWg,30483
 braindecode/models/attn_sleep.py,sha256=F9x4spTtzfiCC1h9UYITmIDQeJW6_2CXTZktZX9R0RE,17950
 braindecode/models/base.py,sha256=yGJgr0f5rD-gJZ5Msw9FzGzVO-x_re-6jOSK8Iht6x4,27923
 braindecode/models/bendr.py,sha256=kqIKtIrgHhIdwYfBFwAJ1g69YbdpA8bs2C2e2GUT0rY,25643
@@ -85,10 +85,10 @@ braindecode/models/sleep_stager_chambon_2018.py,sha256=WvhQGPhBGRBXr07QLKZjHDncv
 braindecode/models/sparcnet.py,sha256=jGK-HMThncg9A0kUqXvREKYieYqOCQzpsYAe1pTub_0,14000
 braindecode/models/sstdpn.py,sha256=zFM-_FROgX3pVHOX5AOln1PNYiSipwfGHVoCZZHDN5s,34219
 braindecode/models/summary.csv,sha256=GGX-0vE6koMLKW-fzBX11OS5tKg9hgY9vPd2e1FGA6w,8508
-braindecode/models/syncnet.py,sha256=3Ry6XrvQG5muGpWOrcVflUvylYFtL_kigXHvsvCzims,8406
+braindecode/models/syncnet.py,sha256=CaQt8do0gf8ydmsOPs1FpNEYyJ_R0CVu_hEDJuMEbHE,8420
 braindecode/models/tcn.py,sha256=ssHaixrCRalGCoP3--wY3ztGwcFQNFta5CeH1nefYas,8237
 braindecode/models/tidnet.py,sha256=IA29vDhcH5LdrZaDDkMu1IAi3mZ39AsS4xA1MYTD-c0,11844
-braindecode/models/tsinception.py,sha256=poBR9lzbaem9MoMY_NqOUBBhSCw9PcEsHk1SkcSU0dY,10364
+braindecode/models/tsinception.py,sha256=hPoDR21ynFscTO31r9aD6vPVm-uPnYCq1ne2ffTA8fI,10374
 braindecode/models/usleep.py,sha256=VaIRbTSuKisD_W6fMwLlY5bNiH41fBnfx0k-iiLHWpw,17316
 braindecode/models/util.py,sha256=QSku9ZlXTyJzMuTLG-VlPPifTe2_6huo7i9-zQnhZho,17728
 braindecode/modules/__init__.py,sha256=NChqITh6yMcagoYFVkQC14aJjCAGkRO8fYQrH_0Zv1k,1977
@@ -119,9 +119,9 @@ braindecode/training/scoring.py,sha256=WRkwqbitA3m_dzRnGp2ZIZPge5Nhx9gAEQhIHzeH4
 braindecode/visualization/__init__.py,sha256=4EER_xHqZIDzEvmgUEm7K1bgNKpyZAIClR9ZCkMuY4M,240
 braindecode/visualization/confusion_matrices.py,sha256=qIWMLEHow5CJ7PhGggD8mnD55Le6xhma9HSzt4R33fc,9509
 braindecode/visualization/gradients.py,sha256=-NuWIlpOeSR2OWGfUl6jn1RqdNPxmiQNLPCckWb8SRE,2330
-braindecode-1.5.0.dev982.dist-info/licenses/LICENSE.txt,sha256=7rg7k6hyj8m9whQ7dpKbqnCssoOEx_Mbtqb4uSOjljE,1525
-braindecode-1.5.0.dev982.dist-info/licenses/NOTICE.txt,sha256=sOxuTbalPxTM8H6VqtvGbXCt_BoOF7JevEYG_knqbm4,620
-braindecode-1.5.0.dev982.dist-info/METADATA,sha256=HyPHY1J4w21v0GHxsAhvIHIWW2gVUkZEt_ZxAeZ6lxA,10099
-braindecode-1.5.0.dev982.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-braindecode-1.5.0.dev982.dist-info/top_level.txt,sha256=pHsWQmSy0uhIez62-HA9j0iaXKvSbUL39ifFRkFnChA,12
-braindecode-1.5.0.dev982.dist-info/RECORD,,
+braindecode-1.5.0.dev984.dist-info/licenses/LICENSE.txt,sha256=7rg7k6hyj8m9whQ7dpKbqnCssoOEx_Mbtqb4uSOjljE,1525
+braindecode-1.5.0.dev984.dist-info/licenses/NOTICE.txt,sha256=sOxuTbalPxTM8H6VqtvGbXCt_BoOF7JevEYG_knqbm4,620
+braindecode-1.5.0.dev984.dist-info/METADATA,sha256=gs4siQJaVjCx8UszxeuPfWYFF_FPk2aagFqBzmIZbTI,10099
+braindecode-1.5.0.dev984.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+braindecode-1.5.0.dev984.dist-info/top_level.txt,sha256=pHsWQmSy0uhIez62-HA9j0iaXKvSbUL39ifFRkFnChA,12
+braindecode-1.5.0.dev984.dist-info/RECORD,,