Braindecode 1.3.2.dev176759409__tar.gz → 1.3.2.dev177700329__tar.gz

{braindecode-1.3.2.dev176759409/braindecode.egg-info → braindecode-1.3.2.dev177700329}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.3.2.dev176759409
+Version: 1.3.2.dev177700329
 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,7 +22,7 @@ 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: torch>=2.2
 Requires-Dist: torchaudio>=2.0
 Requires-Dist: mne>=1.11.0
 Requires-Dist: pandas<3.0.0

{braindecode-1.3.2.dev176759409 → braindecode-1.3.2.dev177700329}/braindecode/models/reve.py

@@ -16,16 +16,7 @@ import torch
 import torch.nn.functional as F
 from einops import rearrange
 from torch import nn
-
-# Safe import for older PyTorch versions (Support for Intel-based Macs)
-try:
-    from torch.nn.attention import SDPBackend, sdpa_kernel
-
-    HAS_SDPA = True
-except ImportError:
-    HAS_SDPA = False
-    SDPBackend = None
-    sdpa_kernel = None
+from torch.nn.attention import SDPBackend, sdpa_kernel
 
 from braindecode.models.base import EEGModuleMixin
 
@@ -539,22 +530,11 @@ class FeedForward(nn.Module):
 
 
 class ClassicalAttention(nn.Module):
-    def __init__(self, heads: int, use_sdpa: bool | None = None):
+    def __init__(self, heads: int, use_sdpa: bool = True):
         super().__init__()
+        self.use_sdpa = use_sdpa
         self.heads = heads
 
-        if use_sdpa is None:
-            self.use_sdpa = HAS_SDPA
-        elif use_sdpa is True and not HAS_SDPA:
-            logger.warning(
-                "SDPA (Scaled Dot Product Attention) was requested, but it is not "
-                "available in your current PyTorch version. Falling back to naive "
-                "implementation. Please upgrade to PyTorch >= 2.2 for SDPA support."
-            )
-            self.use_sdpa = False
-        else:
-            self.use_sdpa = use_sdpa
-
     def forward(self, qkv: torch.Tensor) -> torch.Tensor:
         # Split concatenated QKV into separate tensors
         # qkv shape: (batch, seq_len, 3 * heads * head_dim)

braindecode-1.3.2.dev177700329/braindecode/version.py

@@ -0,0 +1 @@
+__version__ = "1.3.2.dev177700329"

{braindecode-1.3.2.dev176759409 → braindecode-1.3.2.dev177700329/braindecode.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.3.2.dev176759409
+Version: 1.3.2.dev177700329
 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,7 +22,7 @@ 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: torch>=2.2
 Requires-Dist: torchaudio>=2.0
 Requires-Dist: mne>=1.11.0
 Requires-Dist: pandas<3.0.0

{braindecode-1.3.2.dev176759409 → braindecode-1.3.2.dev177700329}/braindecode.egg-info/requires.txt

@@ -1,4 +1,4 @@
-torch>=2.0
+torch>=2.2
 torchaudio>=2.0
 mne>=1.11.0
 pandas<3.0.0

{braindecode-1.3.2.dev176759409 → braindecode-1.3.2.dev177700329}/docs/whats_new.rst

@@ -72,7 +72,6 @@ Enhancements
   pooling (Kostas et al. 2021, Section 2.4), and ``n_chans_pretrained`` parameter with
   max-norm constrained channel projection for fine-tuning pretrained BENDR on datasets
   with arbitrary channel counts (by `Kuntal Kokate`_)
-- Add dynamic SDPA fallback in :class:`braindecode.models.reve.ClassicalAttention` to support PyTorch versions without SDPA (by `GalAshkenazi1`_)
 
 API changes
 ============
@@ -80,7 +79,7 @@ API changes
 
 Requirements
 ============
-- Relaxed PyTorch requirement to >=2.0 to support Intel-based Macs (by `GalAshkenazi1`_)
+- None yet.
 
 Bugs
 =====

{braindecode-1.3.2.dev176759409 → braindecode-1.3.2.dev177700329}/pyproject.toml

@@ -45,7 +45,7 @@ classifiers = [
 readme = "README.rst"
 requires-python = ">=3.11"
 dependencies = [
-    "torch>=2.0",
+    "torch>=2.2",
     "torchaudio>=2.0",
     'mne>=1.11.0',  # we get matplotlib, numpy, pandas, scipy from here
     'pandas<3.0.0',  # to avoid issues with wfdb
@@ -60,15 +60,6 @@ dependencies = [
     'rotary_embedding_torch',
     'pandas<3.0.0',                 # to avoid incompatibility with wfdb https://github.com/MIT-LCP/wfdb-python/issues/560
 ]
-
-[tool.uv]
-required-environments = [
-  "sys_platform == 'darwin' and platform_machine == 'x86_64'",  # Intel Mac
-  "sys_platform == 'darwin' and platform_machine == 'arm64'",   # Apple Silicon Mac (M1/M2/M3)
-  "sys_platform == 'linux' and platform_machine == 'x86_64'",   # Linux 64-bit
-  "sys_platform == 'win32' and platform_machine == 'AMD64'"     # Windows 64-bit
-]
-
 [project.urls]
 homepage = "https://braindecode.org"
 repository = "https://github.com/braindecode/braindecode"

braindecode-1.3.2.dev176759409/braindecode/version.py

@@ -1 +0,0 @@
-__version__ = "1.3.2.dev176759409"