Braindecode 1.3.0.dev168926291__tar.gz → 1.3.0.dev169361694__tar.gz

{braindecode-1.3.0.dev168926291/braindecode.egg-info → braindecode-1.3.0.dev169361694}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.3.0.dev168926291
+Version: 1.3.0.dev169361694
 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>
@@ -14,47 +14,44 @@ Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: Topic :: Software Development :: Build Tools
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3.14
-Requires-Python: >=3.11
+Requires-Python: >=3.10
 Description-Content-Type: text/x-rst
 License-File: LICENSE.txt
 License-File: NOTICE.txt
-Requires-Dist: mne>=1.11.0
+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>=1.3.0
-Requires-Dist: torch>=2.0
-Requires-Dist: torchaudio>=2.0
+Requires-Dist: skorch>=1.2.0
+Requires-Dist: torch<3.0,>=2.0
+Requires-Dist: torchaudio<3.0,>=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
 Requires-Dist: rotary_embedding_torch
 Provides-Extra: moabb
-Requires-Dist: moabb>=1.4.2; extra == "moabb"
+Requires-Dist: moabb>=1.2.0; extra == "moabb"
 Provides-Extra: eegprep
-Requires-Dist: eegprep[eeglabio]>=0.2.23; extra == "eegprep"
+Requires-Dist: eegprep[eeglabio]>=0.1.1; extra == "eegprep"
 Provides-Extra: hub
 Requires-Dist: huggingface_hub[torch]>=0.20.0; extra == "hub"
-Requires-Dist: zarr>=3.0; extra == "hub"
+Requires-Dist: zarr<3.0,>=2.18; extra == "hub"
 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: typing
-Requires-Dist: exca==0.4; extra == "typing"
-Requires-Dist: numpydantic>=1.7; extra == "typing"
 Provides-Extra: docs
 Requires-Dist: sphinx_gallery; extra == "docs"
 Requires-Dist: sphinx_rtd_theme; extra == "docs"
@@ -73,21 +70,14 @@ Requires-Dist: seaborn; extra == "docs"
 Requires-Dist: pre-commit; extra == "docs"
 Requires-Dist: openneuro-py; extra == "docs"
 Requires-Dist: plotly; extra == "docs"
-Requires-Dist: shap; extra == "docs"
-Requires-Dist: nbformat; extra == "docs"
 Provides-Extra: all
 Requires-Dist: braindecode[moabb]; extra == "all"
+Requires-Dist: braindecode[hub]; extra == "all"
 Requires-Dist: braindecode[tests]; extra == "all"
 Requires-Dist: braindecode[docs]; extra == "all"
-Requires-Dist: braindecode[hub]; extra == "all"
 Requires-Dist: braindecode[eegprep]; extra == "all"
-Requires-Dist: braindecode[typing]; extra == "all"
 Dynamic: license-file
 
-.. image:: https://zenodo.org/badge/232335424.svg
-   :target: https://doi.org/10.5281/zenodo.8214376
-   :alt: DOI
-
 .. image:: https://badges.gitter.im/braindecodechat/community.svg
    :alt: Join the chat at https://gitter.im/braindecodechat/community
    :target: https://gitter.im/braindecodechat/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge
@@ -103,7 +93,6 @@ Dynamic: license-file
    :target: https://codecov.io/gh/braindecode/braindecode
    :alt: Code Coverage
 
-
 .. |Braindecode| image:: https://user-images.githubusercontent.com/42702466/177958779-b00628aa-9155-4c51-96d1-d8c345aff575.svg
 .. _braindecode: braindecode.org/
 
@@ -160,15 +149,7 @@ https://gitter.im/braindecodechat/community
 Citing
 ======
 
-If you use Braindecode in scientific work, please cite the software using the Zenodo DOI shown in the badge below:
-
-.. image:: https://zenodo.org/badge/232335424.svg
-   :target: https://doi.org/10.5281/zenodo.8214376
-   :alt: DOI
-
-Additionally, we highly encourage you to cite the article that originally introduced
-the Braindecode library and has served as a foundational reference for many works
-on deep learning with EEG recordings. Please use the following reference:
+If you use this code in a scientific publication, please cite us as:
 
 .. code-block:: bibtex
 

{braindecode-1.3.0.dev168926291 → braindecode-1.3.0.dev169361694}/README.rst

@@ -1,7 +1,3 @@
-.. image:: https://zenodo.org/badge/232335424.svg
-   :target: https://doi.org/10.5281/zenodo.8214376
-   :alt: DOI
-
 .. image:: https://badges.gitter.im/braindecodechat/community.svg
    :alt: Join the chat at https://gitter.im/braindecodechat/community
    :target: https://gitter.im/braindecodechat/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge
@@ -17,7 +13,6 @@
    :target: https://codecov.io/gh/braindecode/braindecode
    :alt: Code Coverage
 
-
 .. |Braindecode| image:: https://user-images.githubusercontent.com/42702466/177958779-b00628aa-9155-4c51-96d1-d8c345aff575.svg
 .. _braindecode: braindecode.org/
 
@@ -74,15 +69,7 @@ https://gitter.im/braindecodechat/community
 Citing
 ======
 
-If you use Braindecode in scientific work, please cite the software using the Zenodo DOI shown in the badge below:
-
-.. image:: https://zenodo.org/badge/232335424.svg
-   :target: https://doi.org/10.5281/zenodo.8214376
-   :alt: DOI
-
-Additionally, we highly encourage you to cite the article that originally introduced
-the Braindecode library and has served as a foundational reference for many works
-on deep learning with EEG recordings. Please use the following reference:
+If you use this code in a scientific publication, please cite us as:
 
 .. code-block:: bibtex
 

{braindecode-1.3.0.dev168926291 → braindecode-1.3.0.dev169361694}/braindecode/augmentation/__init__.py

@@ -5,10 +5,8 @@ Utilities for data augmentation.
 from . import functional
 from .base import AugmentedDataLoader, Compose, IdentityTransform, Transform
 from .transforms import (
-    AmplitudeScale,
     BandstopFilter,
     ChannelsDropout,
-    ChannelsReref,
     ChannelsShuffle,
     ChannelsSymmetry,
     FrequencyShift,
@@ -48,7 +46,5 @@ __all__ = [
     "Mixup",
     "SegmentationReconstruction",
     "MaskEncoding",
-    "AmplitudeScale",
-    "ChannelsReref",
     "functional",
 ]

{braindecode-1.3.0.dev168926291 → braindecode-1.3.0.dev169361694}/braindecode/classifier.py

@@ -8,6 +8,7 @@
 
 import warnings
 
+import numpy as np
 from skorch import NeuralNet
 from skorch.callbacks import EpochScoring
 from skorch.classifier import NeuralNetClassifier
@@ -236,9 +237,13 @@ class EEGClassifier(_EEGNeuralNet, NeuralNetClassifier):
             num_workers=self.get_iterator(X, training=False).loader.num_workers,
         )
 
-    @property
-    def mode(self):
-        return "classification"
+    def _get_n_outputs(self, y, classes):
+        classes_y = np.unique(y)
+        if classes is not None:
+            assert set(classes_y) <= set(classes)
+        else:
+            classes = classes_y
+        return len(classes)
 
     # Only add the 'accuracy' callback if we are not in cropped mode.
     @property

{braindecode-1.3.0.dev168926291 → braindecode-1.3.0.dev169361694}/braindecode/datasets/__init__.py

@@ -11,11 +11,9 @@ from .base import (
 )
 from .bcicomp import BCICompetitionIVDataset4
 from .bids import BIDSDataset, BIDSEpochsDataset
-from .chb_mit import CHBMIT
 from .mne import create_from_mne_epochs, create_from_mne_raw
 from .moabb import BNCI2014_001, HGD, MOABBDataset
 from .nmt import NMT
-from .siena import SIENA
 from .sleep_physio_challe_18 import SleepPhysionetChallenge2018
 from .sleep_physionet import SleepPhysionet
 from .tuh import TUH, TUHAbnormal
@@ -36,9 +34,7 @@ __all__ = [
     "create_from_mne_epochs",
     "TUH",
     "TUHAbnormal",
-    "SIENA",
     "NMT",
-    "CHBMIT",
     "SleepPhysionet",
     "SleepPhysionetChallenge2018",
     "create_from_X_y",

{braindecode-1.3.0.dev168926291 → braindecode-1.3.0.dev169361694}/braindecode/datasets/hub.py

@@ -21,9 +21,18 @@ import mne
 import numpy as np
 import pandas as pd
 import scipy
-from mne._fiff.meas_info import Info
 from mne.utils import _soft_import
 
+# TODO: Simplify this logic in the future with zarr v3+ only
+# Optional imports for Hub functionality
+try:
+    from numcodecs import Blosc, GZip, Zstd
+
+    NUMCODECS_AVAILABLE = True
+except ImportError:
+    NUMCODECS_AVAILABLE = False
+    Blosc = GZip = Zstd = None
+
 if TYPE_CHECKING:
     from .base import BaseDataset
 
@@ -388,8 +397,9 @@ class HubDatasetMixin:
                 f"{output_path} already exists. Set overwrite=True to replace it."
             )
 
-        # Create zarr store (zarr v3 API)
-        root = zarr.open(str(output_path), mode="w")
+        # Create zarr store (zarr v2 API)
+        store = zarr.DirectoryStore(str(output_path))
+        root = zarr.group(store=store, overwrite=False)
 
         # Validate uniformity across all datasets using shared validation
         dataset_type, _, _ = hub_validation.validate_dataset_uniformity(self.datasets)
@@ -434,7 +444,7 @@ class HubDatasetMixin:
                 data = ds.windows.get_data()
                 metadata = ds.windows.metadata
                 description = ds.description
-                info_dict = ds.windows.info.to_json_dict()
+                info_dict = _mne_info_to_dict(ds.windows.info)
                 target_name = ds.target_name if hasattr(ds, "target_name") else None
 
                 # Save using inlined function
@@ -447,7 +457,7 @@ class HubDatasetMixin:
                 raw = ds.raw
                 metadata = ds.metadata
                 description = ds.description
-                info_dict = ds.raw.info.to_json_dict()
+                info_dict = _mne_info_to_dict(ds.raw.info)
                 targets_from = ds.targets_from
                 last_target_only = ds.last_target_only
 
@@ -467,7 +477,7 @@ class HubDatasetMixin:
                 # Get continuous raw data from RawDataset
                 raw = ds.raw
                 description = ds.description
-                info_dict = ds.raw.info.to_json_dict()
+                info_dict = _mne_info_to_dict(ds.raw.info)
                 target_name = ds.target_name if hasattr(ds, "target_name") else None
 
                 # Save using inlined function
@@ -520,8 +530,9 @@ class HubDatasetMixin:
         if not input_path.exists():
             raise FileNotFoundError(f"{input_path} does not exist.")
 
-        # Open zarr store (zarr v3 API)
-        root = zarr.open(str(input_path), mode="r")
+        # Open zarr store (zarr v2 API)
+        store = zarr.DirectoryStore(str(input_path))
+        root = zarr.group(store=store)
 
         n_datasets = root.attrs["n_datasets"]
         dataset_type = root.attrs["dataset_type"]
@@ -543,7 +554,7 @@ class HubDatasetMixin:
                 )
 
                 # Convert to MNE objects and create dataset
-                info = Info.from_json_dict(info_dict)
+                info = _dict_to_mne_info(info_dict)
                 events = np.column_stack(
                     [
                         metadata["i_start_in_trial"].values,
@@ -569,7 +580,7 @@ class HubDatasetMixin:
 
                 # Convert to MNE objects and create dataset
                 # Data is already in continuous format [n_channels, n_timepoints]
-                info = Info.from_json_dict(info_dict)
+                info = _dict_to_mne_info(info_dict)
                 raw = mne.io.RawArray(data, info)
                 ds = EEGWindowsDataset(
                     raw=raw,
@@ -587,7 +598,7 @@ class HubDatasetMixin:
 
                 # Convert to MNE objects and create dataset
                 # Data is in continuous format [n_channels, n_timepoints]
-                info = Info.from_json_dict(info_dict)
+                info = _dict_to_mne_info(info_dict)
                 raw = mne.io.RawArray(data, info)
                 ds = RawDataset(raw, description)
                 if target_name is not None:
@@ -621,22 +632,48 @@ class HubDatasetMixin:
 # =============================================================================
 
 
+# TODO: remove when this MNE is solved https://github.com/mne-tools/mne-python/issues/13487
+def _mne_info_to_dict(info):
+    """Convert MNE Info object to dictionary for JSON serialization."""
+    return {
+        "ch_names": info["ch_names"],
+        "sfreq": float(info["sfreq"]),
+        "ch_types": [str(ch_type) for ch_type in info.get_channel_types()],
+        "lowpass": float(info["lowpass"]) if info["lowpass"] is not None else None,
+        "highpass": float(info["highpass"]) if info["highpass"] is not None else None,
+    }
+
+
+def _dict_to_mne_info(info_dict):
+    """Convert dictionary back to MNE Info object."""
+    info = mne.create_info(
+        ch_names=info_dict["ch_names"],
+        sfreq=info_dict["sfreq"],
+        ch_types=info_dict["ch_types"],
+    )
+
+    # Use _unlock() to set filter info when reconstructing from saved metadata
+    # This is necessary because MNE protects these fields to prevent users from
+    # setting filter parameters without actually filtering the data
+    with info._unlock():
+        if info_dict.get("lowpass") is not None:
+            info["lowpass"] = info_dict["lowpass"]
+        if info_dict.get("highpass") is not None:
+            info["highpass"] = info_dict["highpass"]
+
+    return info
+
+
 def _save_windows_to_zarr(
     grp, data, metadata, description, info, compressor, target_name
 ):
     """Save windowed data to Zarr group (low-level function)."""
     # Save data with chunking for random access
-    # In Zarr v3, use create_array with compressors parameter
-    data_array = data.astype(np.float32)
-
-    # Zarr v3 expects compressors as a list
-    compressors_list = [compressor] if compressor is not None else None
-
-    grp.create_array(
+    grp.create_dataset(
         "data",
-        data=data_array,
-        chunks=(1, data_array.shape[1], data_array.shape[2]),
-        compressors=compressors_list,
+        data=data.astype(np.float32),
+        chunks=(1, data.shape[1], data.shape[2]),
+        compressor=compressor,
     )
 
     # Save metadata
@@ -667,17 +704,11 @@ def _save_eegwindows_to_zarr(
 
     # Save continuous data with chunking optimized for window extraction
     # Chunk size: all channels, 10000 timepoints for efficient random access
-    # In Zarr v3, use create_array with compressors parameter
-    continuous_float = continuous_data.astype(np.float32)
-
-    # Zarr v3 expects compressors as a list
-    compressors_list = [compressor] if compressor is not None else None
-
-    grp.create_array(
+    grp.create_dataset(
         "data",
-        data=continuous_float,
-        chunks=(continuous_float.shape[0], min(10000, continuous_float.shape[1])),
-        compressors=compressors_list,
+        data=continuous_data.astype(np.float32),
+        chunks=(continuous_data.shape[0], min(10000, continuous_data.shape[1])),
+        compressor=compressor,
     )
 
     # Save metadata
@@ -772,17 +803,11 @@ def _save_raw_to_zarr(grp, raw, description, info, target_name, compressor):
 
     # Save continuous data with chunking optimized for efficient access
     # Chunk size: all channels, 10000 timepoints for efficient random access
-    # In Zarr v3, use create_array with compressors parameter
-    continuous_float = continuous_data.astype(np.float32)
-
-    # Zarr v3 expects compressors as a list
-    compressors_list = [compressor] if compressor is not None else None
-
-    grp.create_array(
+    grp.create_dataset(
         "data",
-        data=continuous_float,
-        chunks=(continuous_float.shape[0], min(10000, continuous_float.shape[1])),
-        compressors=compressors_list,
+        data=continuous_data.astype(np.float32),
+        chunks=(continuous_data.shape[0], min(10000, continuous_data.shape[1])),
+        compressor=compressor,
     )
 
     # Save description
@@ -824,34 +849,26 @@ def _load_raw_from_zarr(grp, preload):
 
 
 def _create_compressor(compression, compression_level):
-    """Create a Zarr v3 compressor codec.
-
-    Returns compressor dict compatible with Zarr v3 create_array API.
-
-    Parameters
-    ----------
-    compression : str or None
-        Compression algorithm: "blosc", "zstd", "gzip", or None.
-    compression_level : int
-        Compression level (0-9).
-
-    Returns
-    -------
-    dict or None
-        Compressor configuration dict or None if no compression.
-    """
+    """Create a Zarr compressor object (zarr v2 API)."""
     if zarr is False:
         raise ImportError(
             "Zarr is not installed. Install with: pip install braindecode[hub]"
         )
 
-    if compression is None or compression not in ("blosc", "zstd", "gzip"):
-        return None
-
-    # Map blosc to zstd (Zarr v3 uses zstd as the primary implementation)
-    name = "zstd" if compression == "blosc" else compression
+    if not NUMCODECS_AVAILABLE:
+        raise ImportError(
+            "numcodecs is not installed. Install with: pip install braindecode[hub]"
+        )
 
-    return {"name": name, "configuration": {"level": compression_level}}
+    # Zarr v2 uses numcodecs compressors
+    if compression == "blosc":
+        return Blosc(cname="zstd", clevel=compression_level)
+    elif compression == "zstd":
+        return Zstd(level=compression_level)
+    elif compression == "gzip":
+        return GZip(level=compression_level)
+    else:
+        return None
 
 
 # TODO: improve content

{braindecode-1.3.0.dev168926291 → braindecode-1.3.0.dev169361694}/braindecode/datasets/nmt.py

@@ -32,7 +32,6 @@ from joblib import Parallel, delayed
 from mne.datasets import fetch_dataset
 
 from braindecode.datasets.base import BaseConcatDataset, RawDataset
-from braindecode.datasets.utils import _correct_dataset_path
 
 NMT_URL = "https://zenodo.org/record/10909103/files/NMT.zip"
 NMT_archive_name = "NMT.zip"
@@ -97,34 +96,22 @@ class NMT(BaseConcatDataset):
         preload=False,
         n_jobs=1,
     ):
-        # Convert empty string to None for consistency
-        if path == "":
-            path = None
+        # correct the path if needed
+        if path is not None:
+            list_csv = glob.glob(f"{path}/**/Labels.csv", recursive=True)
+            if isinstance(list_csv, list) and len(list_csv) > 0:
+                path = Path(list_csv[0]).parent
 
-        # Download dataset if not present
-        if path is None:
+        if path is None or len(list_csv) == 0:
             path = fetch_dataset(
                 dataset_params=NMT_dataset_params,
-                path=None,
+                path=Path(path) if path is not None else None,
                 processor="unzip",
                 force_update=False,
             )
             # First time we fetch the dataset, we need to move the files to the
             # correct directory.
-            path = _correct_dataset_path(
-                path, NMT_archive_name, "nmt_scalp_eeg_dataset"
-            )
-        else:
-            # Validate that the provided path is a valid NMT dataset
-            if not Path(f"{path}/Labels.csv").exists():
-                raise ValueError(
-                    f"The provided path {path} does not contain a valid "
-                    "NMT dataset (missing Labels.csv). Please ensure the "
-                    "path points directly to the NMT dataset directory."
-                )
-            path = _correct_dataset_path(
-                path, NMT_archive_name, "nmt_scalp_eeg_dataset"
-            )
+            path = _correct_path(path)
 
         # Get all file paths
         file_paths = glob.glob(
@@ -189,6 +176,39 @@ class NMT(BaseConcatDataset):
         return base_dataset
 
 
+def _correct_path(path: str):
+    """
+    Check if the path is correct and rename the file if needed.
+
+    Parameters
+    ----------
+    path: basestring
+        Path to the file.
+
+    Returns
+    -------
+    path: basestring
+        Corrected path.
+    """
+    if not Path(path).exists():
+        unzip_file_name = f"{NMT_archive_name}.unzip"
+        if (Path(path).parent / unzip_file_name).exists():
+            try:
+                os.rename(
+                    src=Path(path).parent / unzip_file_name,
+                    dst=Path(path),
+                )
+
+            except PermissionError:
+                raise PermissionError(
+                    f"Please rename {Path(path).parent / unzip_file_name}"
+                    + f"manually to {path} and try again."
+                )
+        path = os.path.join(path, "nmt_scalp_eeg_dataset")
+
+    return path
+
+
 def _get_header(*args):
     all_paths = {**_NMT_PATHS}
     return all_paths[args[0]]
@@ -268,33 +288,18 @@ _NMT_PATHS = {
 class _NMTMock(NMT):
     """Mocked class for testing and examples."""
 
-    @mock.patch("pathlib.Path.exists", return_value=True)
-    @mock.patch("braindecode.datasets.nmt._correct_dataset_path")
-    @mock.patch("mne.datasets.fetch_dataset")
-    @mock.patch("pandas.read_csv", new=_fake_pd_read_csv)
-    @mock.patch("mne.io.read_raw_edf", new=_fake_raw)
     @mock.patch("glob.glob", return_value=_NMT_PATHS.keys())
+    @mock.patch("mne.io.read_raw_edf", new=_fake_raw)
+    @mock.patch("pandas.read_csv", new=_fake_pd_read_csv)
     def __init__(
         self,
         mock_glob,
-        mock_fetch,
-        mock_correct_path,
-        mock_path_exists,
         path,
         recording_ids=None,
         target_name="pathological",
         preload=False,
         n_jobs=1,
     ):
-        # Prevent download by providing a dummy path if empty/None
-        if not path:
-            path = "mocked_nmt_path"
-
-        # Mock fetch_dataset to return a valid path without downloading
-        mock_fetch.return_value = path
-        # Mock _correct_dataset_path to return the path as-is
-        mock_correct_path.side_effect = lambda p, *args, **kwargs: p
-
         with warnings.catch_warnings():
             warnings.filterwarnings("ignore", message="Cannot save date file")
             super().__init__(

{braindecode-1.3.0.dev168926291 → braindecode-1.3.0.dev169361694}/braindecode/datautil/__init__.py

@@ -11,7 +11,6 @@ from .serialization import (
     load_concat_dataset,
     save_concat_dataset,
 )
-from .util import infer_signal_properties
 
 
 def __getattr__(name):
@@ -60,5 +59,4 @@ __all__ = [
     "_check_save_dir_empty",
     "match_hemisphere_chans",
     "division_channels_idx",
-    "infer_signal_properties",
 ]

braindecode-1.3.0.dev169361694/braindecode/datautil/util.py

@@ -0,0 +1,41 @@
+# Authors: Robin Schirrmeister <robintibor@gmail.com>
+#
+# License: BSD (3-clause)
+
+
+def ms_to_samples(ms, fs):
+    """
+    Compute milliseconds to number of samples.
+
+    Parameters
+    ----------
+    ms: number
+        Milliseconds
+    fs: number
+        Sampling rate
+
+    Returns
+    -------
+    n_samples: int
+        Number of samples
+
+    """
+    return ms * fs / 1000.0
+
+
+def samples_to_ms(n_samples, fs):
+    """
+    Compute milliseconds to number of samples.
+
+    Parameters
+    ----------
+    n_samples: number
+        Number of samples
+    fs: number
+        Sampling rate
+
+    Returns
+    -------
+    milliseconds: int
+    """
+    return n_samples * 1000.0 / fs