scitex 2.5.0__py3-none-any.whl → 2.7.0__py3-none-any.whl

scitex/decorators/_pandas_fn.py

@@ -54,9 +54,9 @@ def pandas_fn(func: Callable) -> Callable:
                 except:
                     # If list can't be converted to DataFrame, return as is
                     return data
-            elif hasattr(data, '__class__') and data.__class__.__name__ == 'Tensor':
+            elif hasattr(data, "__class__") and data.__class__.__name__ == "Tensor":
                 return pd.DataFrame(data.detach().cpu().numpy())
-            elif hasattr(data, '__class__') and data.__class__.__name__ == 'DataArray':
+            elif hasattr(data, "__class__") and data.__class__.__name__ == "DataArray":
                 return pd.DataFrame(data.values)
             elif isinstance(data, (int, float, str)):
                 # Don't convert scalars to DataFrames
@@ -73,6 +73,7 @@ def pandas_fn(func: Callable) -> Callable:
 
         # Skip strict assertion for certain types
         import pandas as pd
+
         validated_args = []
         for arg_index, arg in enumerate(converted_args):
             if isinstance(arg, pd.DataFrame):
@@ -80,7 +81,9 @@ def pandas_fn(func: Callable) -> Callable:
             elif isinstance(arg, (int, float, str, type(None), pd.Series)):
                 # Pass through scalars, strings, Series, and None unchanged
                 validated_args.append(arg)
-            elif isinstance(arg, list) and all(isinstance(item, pd.DataFrame) for item in arg):
+            elif isinstance(arg, list) and all(
+                isinstance(item, pd.DataFrame) for item in arg
+            ):
                 # List of DataFrames - pass through as is
                 validated_args.append(arg)
             else:
@@ -95,14 +98,19 @@ def pandas_fn(func: Callable) -> Callable:
 
         # Convert results back to original input types
         import pandas as pd
+
         if isinstance(results, pd.DataFrame):
             if original_object is not None:
                 if isinstance(original_object, list):
                     return results.values.tolist()
                 elif isinstance(original_object, np.ndarray):
                     return results.values
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'Tensor':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "Tensor"
+                ):
                     import torch
+
                     return torch.tensor(results.values)
                 elif isinstance(original_object, pd.Series):
                     return (
@@ -110,8 +118,12 @@ def pandas_fn(func: Callable) -> Callable:
                         if results.shape[1] > 0
                         else pd.Series()
                     )
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'DataArray':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "DataArray"
+                ):
                     import xarray as xr
+
                     return xr.DataArray(results.values)
             return results
 

scitex/decorators/_signal_fn.py

@@ -53,26 +53,40 @@ def signal_fn(func: Callable) -> Callable:
 
         # Convert results back to original input types
         import torch
+
         if isinstance(results, torch.Tensor):
             if original_object is not None:
                 if isinstance(original_object, list):
                     return results.detach().cpu().numpy().tolist()
                 elif isinstance(original_object, np.ndarray):
                     return results.detach().cpu().numpy()
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'DataFrame':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "DataFrame"
+                ):
                     import pandas as pd
+
                     return pd.DataFrame(results.detach().cpu().numpy())
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'Series':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "Series"
+                ):
                     import pandas as pd
+
                     return pd.Series(results.detach().cpu().numpy().flatten())
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'DataArray':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "DataArray"
+                ):
                     import xarray as xr
+
                     return xr.DataArray(results.detach().cpu().numpy())
             return results
 
         # Handle tuple returns (e.g., (signal, frequencies))
         elif isinstance(results, tuple):
             import torch
+
             converted_results = []
             for r in results:
                 if isinstance(r, torch.Tensor):

scitex/decorators/_torch_fn.py

@@ -20,54 +20,55 @@ from ._converters import _return_always, is_nested_decorator, to_torch
 
 def torch_fn(func: Callable) -> Callable:
     """Decorator for PyTorch function compatibility.
-    
+
     Automatically converts inputs to PyTorch tensors and handles various data types
     gracefully. Preserves the original input type in the output.
-    
+
     Features
     --------
     - Converts inputs to PyTorch tensors
-    - Preserves scalar parameters (int, float, bool, str) 
+    - Preserves scalar parameters (int, float, bool, str)
     - Preserves dimension tuples like dim=(0, 1)
     - Handles nested lists/tuples gracefully
     - Automatically converts axis to dim for torch functions
     - Applies device="cuda" if available
     - Returns output in same type as input (numpy->numpy, pandas->pandas, etc.)
-    
+
     Parameters
     ----------
     func : Callable
         The function to decorate
-        
+
     Returns
     -------
     Callable
         The decorated function
-        
+
     Examples
     --------
     >>> @torch_fn
     ... def mean_squared(x, dim=None):
     ...     return (x ** 2).mean(dim=dim)
-    >>> 
+    >>>
     >>> # Works with numpy arrays
     >>> result = mean_squared(np.array([1, 2, 3]))
-    >>> 
+    >>>
     >>> # Works with nested lists
     >>> result = mean_squared([[1, 2], [3, 4]])
-    >>> 
+    >>>
     >>> # Preserves dimension tuples
     >>> result = mean_squared(data, dim=(0, 1))
-    
+
     Notes
     -----
     For optimal performance with batch processing, apply torch_fn before batch_fn:
     @batch_fn
     @torch_fn
     def my_function(x): ...
-    
+
     Or use auto-ordering to handle this automatically.
     """
+
     @wraps(func)
     def wrapper(*args: _Any, **kwargs: _Any) -> _Any:
         # Skip conversion if already in a nested decorator context
@@ -88,6 +89,7 @@ def torch_fn(func: Callable) -> Callable:
         # Skip strict assertion for certain types that may not convert to tensors
         # Instead, convert what we can and pass through what we can't
         import torch
+
         validated_args = []
         for arg_index, arg in enumerate(converted_args):
             if isinstance(arg, torch.Tensor):
@@ -95,7 +97,9 @@ def torch_fn(func: Callable) -> Callable:
             elif isinstance(arg, (int, float, str, type(None))):
                 # Pass through scalars and strings unchanged
                 validated_args.append(arg)
-            elif isinstance(arg, list) and all(isinstance(item, torch.Tensor) for item in arg):
+            elif isinstance(arg, list) and all(
+                isinstance(item, torch.Tensor) for item in arg
+            ):
                 # List of tensors - pass through as is
                 validated_args.append(arg)
             else:
@@ -110,20 +114,33 @@ def torch_fn(func: Callable) -> Callable:
 
         # Convert results back to original input types
         import torch
+
         if isinstance(results, torch.Tensor):
             if original_object is not None:
                 if isinstance(original_object, list):
                     return results.detach().cpu().numpy().tolist()
                 elif isinstance(original_object, np.ndarray):
                     return results.detach().cpu().numpy()
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'DataFrame':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "DataFrame"
+                ):
                     import pandas as pd
+
                     return pd.DataFrame(results.detach().cpu().numpy())
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'Series':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "Series"
+                ):
                     import pandas as pd
+
                     return pd.Series(results.detach().cpu().numpy().flatten())
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'DataArray':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "DataArray"
+                ):
                     import xarray as xr
+
                     return xr.DataArray(results.detach().cpu().numpy())
             return results
 

scitex/decorators/_xarray_fn.py

@@ -43,11 +43,11 @@ def xarray_fn(func: Callable) -> Callable:
                 return xr.DataArray(data)
             elif isinstance(data, list):
                 return xr.DataArray(data)
-            elif hasattr(data, '__class__') and data.__class__.__name__ == 'Tensor':
+            elif hasattr(data, "__class__") and data.__class__.__name__ == "Tensor":
                 return xr.DataArray(data.detach().cpu().numpy())
-            elif hasattr(data, '__class__') and data.__class__.__name__ == 'DataFrame':
+            elif hasattr(data, "__class__") and data.__class__.__name__ == "DataFrame":
                 return xr.DataArray(data.values)
-            elif hasattr(data, '__class__') and data.__class__.__name__ == 'Series':
+            elif hasattr(data, "__class__") and data.__class__.__name__ == "Series":
                 return xr.DataArray(data.values)
             else:
                 return xr.DataArray([data])
@@ -57,29 +57,43 @@ def xarray_fn(func: Callable) -> Callable:
 
         # Assertion to ensure all args are converted to xarray DataArrays
         import xarray as xr
+
         for arg_index, arg in enumerate(converted_args):
-            assert isinstance(
-                arg, xr.DataArray
-            ), f"Argument {arg_index} not converted to DataArray: {type(arg)}"
+            assert isinstance(arg, xr.DataArray), (
+                f"Argument {arg_index} not converted to DataArray: {type(arg)}"
+            )
 
         results = func(*converted_args, **converted_kwargs)
 
         # Convert results back to original input types
         import xarray as xr
+
         if isinstance(results, xr.DataArray):
             if original_object is not None:
                 if isinstance(original_object, list):
                     return results.values.tolist()
                 elif isinstance(original_object, np.ndarray):
                     return results.values
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'Tensor':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "Tensor"
+                ):
                     import torch
+
                     return torch.tensor(results.values)
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'DataFrame':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "DataFrame"
+                ):
                     import pandas as pd
+
                     return pd.DataFrame(results.values)
-                elif hasattr(original_object, '__class__') and original_object.__class__.__name__ == 'Series':
+                elif (
+                    hasattr(original_object, "__class__")
+                    and original_object.__class__.__name__ == "Series"
+                ):
                     import pandas as pd
+
                     return pd.Series(results.values.flatten())
             return results
 

scitex/dev/_analyze_code_flow.py

@@ -97,7 +97,6 @@ class CodeFlowAnalyzer:
         sequence_orig = self.sequence
 
         if isinstance(node, (ast.FunctionDef, ast.ClassDef)):
-
             if node.name not in self.skip_functions:
                 # Track all function definitions
                 self.execution_flow.append((depth, node.name, self.sequence))
@@ -158,7 +157,6 @@ class CodeFlowAnalyzer:
         filtered_flow = []
 
         for depth, call, seq in self.execution_flow:
-
             # Start skipping when encountering private method
             if call.startswith(("_", "self._")):
                 skip_until_depth = depth

scitex/dict/_DotDict.py

@@ -99,7 +99,7 @@ class DotDict:
         result = {}
         for key, value in self._data.items():
             # Skip private keys (starting with _) unless explicitly requested
-            if not include_private and isinstance(key, str) and key.startswith('_'):
+            if not include_private and isinstance(key, str) and key.startswith("_"):
                 continue
             if isinstance(value, DotDict):
                 value = value.to_dict(include_private=include_private)
@@ -125,9 +125,7 @@ class DotDict:
 
         try:
             # Use the internal _data for representation
-            return json.dumps(
-                self.to_dict(), indent=4, default=default_handler
-            )
+            return json.dumps(self.to_dict(), indent=4, default=default_handler)
         except TypeError as e:
             # Fallback if default_handler still fails (e.g., complex recursion)
             return f"<DotDict object at {hex(id(self))}, contains: {list(self._data.keys())}> Error: {e}"
@@ -140,7 +138,9 @@ class DotDict:
         """
         # Use pprint.pformat for nice formatting
         # Convert to regular dict recursively, hiding private keys
-        return _pprint.pformat(self.to_dict(include_private=False), indent=2, width=80, compact=False)
+        return _pprint.pformat(
+            self.to_dict(include_private=False), indent=2, width=80, compact=False
+        )
 
     def _repr_pretty_(self, p, cycle):
         """
@@ -153,17 +153,19 @@ class DotDict:
             cycle: Boolean indicating if we're in a reference cycle
         """
         if cycle:
-            p.text('DotDict(...)')
+            p.text("DotDict(...)")
         else:
-            with p.group(8, 'DotDict(', ')'):
+            with p.group(8, "DotDict(", ")"):
                 if self._data:
                     # Filter out private keys for display
                     public_data = self.to_dict(include_private=False)
                     p.pretty(public_data)
                 else:
-                    p.text('{}')
+                    p.text("{}")
 
-    def pformat(self, indent=2, width=80, depth=None, compact=False, include_private=False):
+    def pformat(
+        self, indent=2, width=80, depth=None, compact=False, include_private=False
+    ):
         """
         Return a pretty-formatted string representation of the DotDict.
 
@@ -182,7 +184,7 @@ class DotDict:
             indent=indent,
             width=width,
             depth=depth,
-            compact=compact
+            compact=compact,
         )
 
     def __len__(self):
@@ -247,9 +249,7 @@ class DotDict:
         """
         # Mimic dict.pop behavior with optional default
         if len(args) > 1:
-            raise TypeError(
-                f"pop expected at most 2 arguments, got {1 + len(args)}"
-            )
+            raise TypeError(f"pop expected at most 2 arguments, got {1 + len(args)}")
         if key not in self._data:
             if args:
                 return args[0]  # Return default if provided
@@ -320,9 +320,7 @@ class DotDict:
             if len(self._data) == 0:
                 return 0 < other
             # If single numeric value in total field, use it
-            if "total" in self._data and isinstance(
-                self._data["total"], (int, float)
-            ):
+            if "total" in self._data and isinstance(self._data["total"], (int, float)):
                 return self._data["total"] < other
             # Otherwise not comparable
             return NotImplemented
@@ -339,9 +337,7 @@ class DotDict:
             if len(self._data) == 0:
                 return 0 > other
             # If single numeric value in total field, use it
-            if "total" in self._data and isinstance(
-                self._data["total"], (int, float)
-            ):
+            if "total" in self._data and isinstance(self._data["total"], (int, float)):
                 return self._data["total"] > other
             # Otherwise not comparable
             return NotImplemented

scitex/dict/_flatten.py

@@ -16,4 +16,5 @@ def flatten(nested_dict, parent_key="", sep="_"):
             items.append((new_key, value))
     return dict(items)
 
+
 # EOF

scitex/dict/_listed_dict.py

@@ -38,4 +38,5 @@ def listed_dict(keys=None):
             dict_list[k] = []
     return dict_list
 
+
 # EOF

scitex/dict/_pop_keys.py

@@ -32,4 +32,5 @@ def pop_keys(keys_list, keys_to_pop):
     keys_remainded_list = list(np.array(keys_list)[list(indi_to_remain)])
     return keys_remainded_list
 
+
 # EOF

scitex/dict/_replace.py

@@ -8,4 +8,5 @@ def replace(string, dict):
         string = string.replace(k, v)
     return string
 
+
 # EOF

scitex/dict/_safe_merge.py

@@ -49,4 +49,5 @@ def safe_merge(*dicts: Dict[_Any, _Any]) -> Dict[_Any, _Any]:
     except Exception as error:
         raise ValueError(f"Dictionary merge failed: {str(error)}")
 
+
 # EOF

scitex/dict/_to_str.py

@@ -28,8 +28,7 @@ def to_str(dictionary, delimiter="_"):
     str
         A string representation of the input dictionary.
     """
-    return delimiter.join(
-        f"{key}-{value}" for key, value in dictionary.items()
-    )
+    return delimiter.join(f"{key}-{value}" for key, value in dictionary.items())
+
 
 # EOF

scitex/dsp/__init__.py

@@ -6,7 +6,14 @@ import warnings
 # Core imports that should always work
 from ._crop import crop
 from ._demo_sig import demo_sig
-from ._detect_ripples import detect_ripples, _preprocess, _find_events, _drop_ripples_at_edges, _calc_relative_peak_position, _sort_columns
+from ._detect_ripples import (
+    detect_ripples,
+    _preprocess,
+    _find_events,
+    _drop_ripples_at_edges,
+    _calc_relative_peak_position,
+    _sort_columns,
+)
 from ._ensure_3d import ensure_3d
 from ._hilbert import hilbert
 from ._modulation_index import modulation_index, _reshape
@@ -24,12 +31,13 @@ from . import params
 # Try to import audio-related functions that require PortAudio
 try:
     from ._listen import list_and_select_device
+
     _audio_available = True
 except (ImportError, OSError) as e:
     warnings.warn(
         "Audio functionality unavailable: PortAudio library not found. "
         "Install PortAudio to use audio features (e.g., sudo apt-get install portaudio19-dev)",
-        ImportWarning
+        ImportWarning,
     )
     list_and_select_device = None
     _audio_available = False
@@ -37,11 +45,12 @@ except (ImportError, OSError) as e:
 # Try to import MNE-related functions
 try:
     from ._mne import get_eeg_pos
+
     _mne_available = True
 except ImportError:
     warnings.warn(
         "MNE functionality unavailable. Install MNE-Python to use EEG position features.",
-        ImportWarning
+        ImportWarning,
     )
     get_eeg_pos = None
     _mne_available = False
@@ -69,4 +78,4 @@ __all__ = [
     "to_segments",
     "to_sktime_df",
     "wavelet",
-]
+]

scitex/dsp/_crop.py

@@ -115,7 +115,9 @@ if __name__ == "__main__":
     # parser.add_argument('--flag', '-f', action='store_true', default=False, help='')
     # args = parser.parse_args()
     # Main
-    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(sys, plt, verbose=False)
+    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(
+        sys, plt, verbose=False
+    )
     main()
     scitex.session.close(CONFIG, verbose=False, notify=False)
 

scitex/dsp/_detect_ripples.py

@@ -205,7 +205,9 @@ if __name__ == "__main__":
     # parser.add_argument('--flag', '-f', action='store_true', default=False, help='')
     # args = parser.parse_args()
     # Main
-    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(sys, plt, verbose=False)
+    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(
+        sys, plt, verbose=False
+    )
     main()
     scitex.session.close(CONFIG, verbose=False, notify=False)
 

scitex/dsp/_modulation_index.py

@@ -35,7 +35,9 @@ if __name__ == "__main__":
     import scitex
 
     # Start
-    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(sys, plt, fig_scale=3)
+    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(
+        sys, plt, fig_scale=3
+    )
 
     # Parameters
     FS = 512

scitex/dsp/_time.py

@@ -30,7 +30,9 @@ if __name__ == "__main__":
     # parser.add_argument('--flag', '-f', action='store_true', default=False, help='')
     # args = parser.parse_args()
     # Main
-    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(sys, plt, verbose=False)
+    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(
+        sys, plt, verbose=False
+    )
     main()
     scitex.session.close(CONFIG, verbose=False, notify=False)
 

scitex/dsp/_wavelet.py

@@ -5,7 +5,6 @@
 
 """scitex.dsp.wavelet function"""
 
-
 from scitex.decorators import batch_fn, signal_fn
 from scitex.nn._Wavelet import Wavelet
 import scitex

scitex/dsp/example.py

@@ -74,7 +74,6 @@ def plot_signals(plt, sigs, sig_type):
     i_batch = 0
     i_ch = 0
     for ax, (i_col, col) in zip(axes, enumerate(sigs.columns)):
-
         if col == "hilbert_amp":  # add the original signal to the ax
             _col = "orig"
             (
@@ -115,7 +114,6 @@ def plot_signals(plt, sigs, sig_type):
 
 
 def plot_wavelet(plt, sigs, sig_col, sig_type):
-
     xx, tt, fs = sigs[sig_col]
     # if sig_type == "tensorpac":
     #     xx = xx[:, :, 0]
@@ -160,7 +158,6 @@ def plot_wavelet(plt, sigs, sig_col, sig_type):
 
 
 def plot_psd(plt, sigs, sig_col, sig_type):
-
     xx, tt, fs = sigs[sig_col]
 
     # if sig_type == "tensorpac":
@@ -200,7 +197,6 @@ def plot_psd(plt, sigs, sig_col, sig_type):
 
 
 if __name__ == "__main__":
-
     # Parameters
     T_SEC = 4
     SIG_TYPES = [
@@ -237,7 +233,6 @@ if __name__ == "__main__":
 
         # Plots wavelet coefficients and PSD
         for sig_col in sigs.columns:
-
             if "hilbert" in sig_col:
                 continue
 

scitex/dsp/filt.py

@@ -15,6 +15,7 @@ from scitex.decorators import signal_fn
 @signal_fn
 def gauss(x, sigma, t=None):
     from scitex.nn._Filters import GaussianFilter
+
     return GaussianFilter(sigma)(x, t=t)
 
 
@@ -32,18 +33,21 @@ def bandpass(x, fs, bands, t=None):
 @signal_fn
 def bandstop(x, fs, bands, t=None):
     from scitex.nn._Filters import BandStopFilter
+
     return BandStopFilter(bands, fs, x.shape[-1])(x, t=t)
 
 
 @signal_fn
 def lowpass(x, fs, cutoffs_hz, t=None):
     from scitex.nn._Filters import LowPassFilter
+
     return LowPassFilter(cutoffs_hz, fs, x.shape[-1])(x, t=t)
 
 
 @signal_fn
 def highpass(x, fs, cutoffs_hz, t=None):
     from scitex.nn._Filters import HighPassFilter
+
     return HighPassFilter(cutoffs_hz, fs, x.shape[-1])(x, t=t)
 
 

scitex/dsp/utils/__init__.py

@@ -1,7 +1,10 @@
 #!/usr/bin/env python3
 """Scitex utils module."""
 
-from ._differential_bandpass_filters import build_bandpass_filters, init_bandpass_filters
+from ._differential_bandpass_filters import (
+    build_bandpass_filters,
+    init_bandpass_filters,
+)
 from ._ensure_3d import ensure_3d
 from ._ensure_even_len import ensure_even_len
 from ._zero_pad import zero_pad

scitex/dsp/utils/pac.py

@@ -110,9 +110,9 @@ def plot_PAC_scitex_vs_tensorpac(pac_scitex, pac_tp, freqs_pha, freqs_amp):
 def plot_pac_scitex_vs_tensorpac(pac_scitex, pac_tp, freqs_pha, freqs_amp):
     """
     Plot comparison between SciTeX and Tensorpac phase-amplitude coupling results.
-    
+
     This is an alias for plot_PAC_scitex_vs_tensorpac with snake_case naming.
-    
+
     Parameters
     ----------
     pac_scitex : array-like
@@ -123,7 +123,7 @@ def plot_pac_scitex_vs_tensorpac(pac_scitex, pac_tp, freqs_pha, freqs_amp):
         Phase frequencies
     freqs_amp : array-like
         Amplitude frequencies
-        
+
     Returns
     -------
     fig : matplotlib.figure.Figure