Braindecode 1.8.0.dev1115__tar.gz → 1.8.0.dev1125__tar.gz

{braindecode-1.8.0.dev1115/braindecode.egg-info → braindecode-1.8.0.dev1125}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.8.0.dev1115
+Version: 1.8.0.dev1125
 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.8.0.dev1115 → braindecode-1.8.0.dev1125}/braindecode/models/base.py

@@ -198,34 +198,7 @@ class EEGModuleMixin(_BaseHubMixin, metaclass=_BraindecodeDocstringMeta):
     See :ref:`load-pretrained-models` for a complete tutorial.
     """
 
-    #: Attributes that :func:`torch.jit.script` must not introspect. The
-    #: signal-related properties raise :class:`ValueError` when their value was
-    #: neither given nor inferable, and ``mapping`` carries a postponed
-    #: annotation TorchScript cannot resolve. Either one aborts scripting
-    #: before ``forward`` is ever compiled.
-    __jit_ignored_attributes__ = [
-        *sorted(_EEG_PARAMS),
-        "_chs_info",
-        "input_shape",
-        "mapping",
-    ]
-    #: Rich MNE channel dictionaries are intentionally unavailable in scripted
-    #: forwards; they remain unchanged on eager models and in saved configs.
-    __jit_unused_properties__ = ["chs_info"]
-
     def __init_subclass__(cls, **kwargs):
-        # TorchScript only honours ``__jit_ignored_attributes__`` for
-        # properties defined directly on the concrete class: it collects them
-        # with ``vars(type(module))``, which skips inherited ones. Rebinding
-        # the very same property objects on each subclass makes them visible
-        # there without changing any runtime behaviour.
-        for name in cls.__jit_ignored_attributes__:
-            if name in cls.__dict__:
-                continue
-            prop = getattr(cls, name, None)
-            if isinstance(prop, property):
-                setattr(cls, name, prop)
-
         # Append model-specific Hub integration notes to the docstring.
         # This runs before the metaclass __init__, so the Hub notes will
         # be included in the docstring that the metaclass processes.
@@ -313,12 +286,6 @@ class EEGModuleMixin(_BaseHubMixin, metaclass=_BraindecodeDocstringMeta):
         self._chs_info = chs_info  # type: ignore[assignment]
         self._n_outputs = n_outputs  # type: ignore[assignment]
         self._n_chans = n_chans  # type: ignore[assignment]
-        # TorchScript cannot represent the rich MNE dictionaries in _chs_info.
-        # Keep the original eager state above and expose only the derived scalar
-        # to scripted signal-property getters.
-        self._n_chans_for_jit = (
-            len(chs_info) if n_chans is None and chs_info is not None else n_chans
-        )
         self._n_times = n_times  # type: ignore[assignment]
         self._sfreq = sfreq  # type: ignore[assignment]
 
@@ -343,12 +310,6 @@ class EEGModuleMixin(_BaseHubMixin, metaclass=_BraindecodeDocstringMeta):
 
     @property
     def n_chans(self) -> int:
-        if torch.jit.is_scripting():
-            if self._n_chans_for_jit is None:
-                raise ValueError(
-                    "n_chans could not be inferred. Either specify n_chans or chs_info."
-                )
-            return self._n_chans_for_jit
         if self._n_chans is None and self._chs_info is not None:
             return len(self._chs_info)
         elif self._n_chans is None:
@@ -358,7 +319,7 @@ class EEGModuleMixin(_BaseHubMixin, metaclass=_BraindecodeDocstringMeta):
         return self._n_chans
 
     @property
-    def chs_info(self) -> list[dict]:
+    def chs_info(self) -> list[str]:
         if self._chs_info is None:
             raise ValueError("chs_info not specified.")
         return self._chs_info

braindecode-1.8.0.dev1125/braindecode/version.py

@@ -0,0 +1 @@
+__version__ = "1.8.0.dev1125"

{braindecode-1.8.0.dev1115 → braindecode-1.8.0.dev1125/braindecode.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.8.0.dev1115
+Version: 1.8.0.dev1125
 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.8.0.dev1115 → braindecode-1.8.0.dev1125}/docs/whats_new.rst

@@ -28,14 +28,6 @@ Current 1.8.0 (GitHub)
 Enhancements
 ============
 
-- Models with TorchScript-compatible forward paths can now be passed straight
-  to :func:`torch.jit.script`, without first being rebuilt as a plain
-  :class:`torch.nn.Module`.
-  :class:`braindecode.models.base.EEGModuleMixin` hides its signal-related
-  properties and the ``mapping`` attribute from TorchScript introspection, so
-  scripting reaches ``forward`` instead of failing on the class attributes.
-  (:gh:`1115` by `Aditya Singh`_)
-
 - Add a transfer-learning tutorial illustrating the TUAB-to-NMT pathology
   workflow with offline synthetic recordings
   (:gh:`580` by `Mohammad Javad D`_ and `Bruno Aristimunha`_).
@@ -1620,4 +1612,3 @@ Authors
 .. _Fashad Ahmed: https://github.com/Fashad-Ahmed
 .. _Bhargav Kowshik: https://github.com/bkowshik
 .. _Jon Huml: https://github.com/jonathanhuml
-.. _Aditya Singh: https://github.com/adityasingh2400

braindecode-1.8.0.dev1115/braindecode/version.py

@@ -1 +0,0 @@
-__version__ = "1.8.0.dev1115"