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

scitex/linalg/__init__.py

@@ -8,7 +8,7 @@ from ._misc import cosine, nannorm, rebase_a_vec, three_line_lengths_to_coords
 __all__ = [
     "euclidean_distance",
     "cdist",
-    "edist", 
+    "edist",
     "geometric_median",
     "cosine",
     "nannorm",

scitex/linalg/_geometric_median.py

@@ -23,7 +23,6 @@ from scitex.decorators import torch_fn
 
 @torch_fn
 def geometric_median(xx, dim=-1):
-
     # Ensure dim is a positive index
     if dim < 0:
         dim = xx.ndim + dim
@@ -52,7 +51,7 @@ if __name__ == "__main__":
     import sys
     import matplotlib.pyplot as plt
     import scitex
-    
+
     # # Argument Parser
     # import argparse
     # parser = argparse.ArgumentParser(description='')
@@ -61,7 +60,9 @@ if __name__ == "__main__":
     # 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/logging/_Tee.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/logging/_Tee.py"
-)
+
+__FILE__ = "./src/scitex/logging/_Tee.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -39,6 +38,8 @@ from scitex.str._clean_path import clean_path
 from scitex.str._printc import printc
 
 """Functions & Classes"""
+
+
 def _get_logger():
     """Get logger lazily to avoid circular import during module initialization."""
     from scitex import logging
@@ -111,10 +112,7 @@ class Tee:
         if hasattr(self, "_log_file") and self._log_file is not None:
             try:
                 # Check if the file object is still valid
-                if (
-                    hasattr(self._log_file, "closed")
-                    and not self._log_file.closed
-                ):
+                if hasattr(self._log_file, "closed") and not self._log_file.closed:
                     self.close()
             except Exception:
                 # Silently ignore exceptions during cleanup

scitex/logging/__init__.py

@@ -5,6 +5,7 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
+
 __FILE__ = __file__
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
@@ -43,12 +44,13 @@ warnings.warn(
     "scitex.logging is deprecated. Use scitex.logging instead. "
     "The scitex.logging module will be removed in a future version.",
     DeprecationWarning,
-    stacklevel=2
+    stacklevel=2,
 )
 
 import logging as _logging
 
 from ._Tee import Tee, tee
+
 # Import modular components
 from ._levels import SUCCESS, FAIL, DEBUG, INFO, WARNING, ERROR, CRITICAL
 from ._logger import SciTeXLogger, setup_logger_class
@@ -60,13 +62,13 @@ from ._config import (
     enable_file_logging,
     is_file_logging_enabled,
     configure,
-    get_log_path
+    get_log_path,
 )
 from ._print_capture import (
     PrintCapture,
     enable_print_capture,
     disable_print_capture,
-    is_print_capture_enabled
+    is_print_capture_enabled,
 )
 from ._context import log_to_file
 
@@ -98,25 +100,22 @@ configure(level=level, enable_file=True, enable_console=True, capture_prints=Fal
 # Export only essential public functions - minimal API
 __all__ = [
     # Core logging functions (most commonly used)
-    'getLogger',
-
+    "getLogger",
     # Log levels
-    'DEBUG',
-    'INFO',
-    'WARNING',
-    'ERROR',
-    'CRITICAL',
-    'SUCCESS',
-    'FAIL',
-
+    "DEBUG",
+    "INFO",
+    "WARNING",
+    "ERROR",
+    "CRITICAL",
+    "SUCCESS",
+    "FAIL",
     # Configuration (minimal set)
-    'configure',
-    'get_log_path',
-    'Tee',
-    'tee',
-
+    "configure",
+    "get_log_path",
+    "Tee",
+    "tee",
     # Context managers
-    'log_to_file',
+    "log_to_file",
 ]
 
 # EOF

scitex/logging/_config.py

@@ -5,6 +5,7 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
+
 __FILE__ = __file__
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
@@ -128,7 +129,9 @@ def configure(
 
     # Configure basic logging with our handlers
     logging.basicConfig(
-        level=level, handlers=handlers, force=True  # Force reconfiguration
+        level=level,
+        handlers=handlers,
+        force=True,  # Force reconfiguration
     )
 
     # Enable print capture if requested

scitex/logging/_context.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/logging/_context.py"
-)
+
+__FILE__ = "./src/scitex/logging/_context.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -29,7 +28,7 @@ from ._formatters import SciTeXFileFormatter
 def log_to_file(
     file_path: Union[str, Path],
     level: int = _logging.DEBUG,
-    mode: str = 'w',
+    mode: str = "w",
     formatter: Optional[_logging.Formatter] = None,
 ):
     """Context manager to temporarily log all output to a specific file.
@@ -72,6 +71,7 @@ def log_to_file(
     def _log_info():
         try:
             from scitex import logging
+
             logger = logging.getLogger(__name__)
             logger.info(f"Logging to: {file_path}")
         except:
@@ -90,6 +90,7 @@ def log_to_file(
         def _log_saved():
             try:
                 from scitex import logging
+
                 logger = logging.getLogger(__name__)
                 logger.info(f"Log saved: {file_path}")
             except:
@@ -98,6 +99,6 @@ def log_to_file(
         _log_saved()
 
 
-__all__ = ['log_to_file']
+__all__ = ["log_to_file"]
 
 # EOF

scitex/logging/_formatters.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/logging/_formatters.py"
-)
+
+__FILE__ = "./src/scitex/logging/_formatters.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 

scitex/logging/_handlers.py

@@ -4,10 +4,11 @@
 
 import logging
 import logging.handlers
-import os
 from datetime import datetime
 from pathlib import Path
 
+from scitex.config import get_scitex_dir
+
 from ._formatters import SciTeXConsoleFormatter, SciTeXFileFormatter
 
 
@@ -19,9 +20,11 @@ def create_console_handler(level=logging.INFO):
     return handler
 
 
-def create_file_handler(log_file_path, level=logging.INFO, max_bytes=10*1024*1024, backup_count=5):
+def create_file_handler(
+    log_file_path, level=logging.INFO, max_bytes=10 * 1024 * 1024, backup_count=5
+):
     """Create a rotating file handler for log files.
-    
+
     Args:
         log_file_path: Path to the log file
         level: Log level for the handler
@@ -31,13 +34,10 @@ def create_file_handler(log_file_path, level=logging.INFO, max_bytes=10*1024*102
     # Ensure the log directory exists
     log_dir = Path(log_file_path).parent
     log_dir.mkdir(parents=True, exist_ok=True)
-    
+
     # Use RotatingFileHandler to prevent log files from growing too large
     handler = logging.handlers.RotatingFileHandler(
-        log_file_path,
-        maxBytes=max_bytes,
-        backupCount=backup_count,
-        encoding='utf-8'
+        log_file_path, maxBytes=max_bytes, backupCount=backup_count, encoding="utf-8"
     )
     handler.setLevel(level)
     handler.setFormatter(SciTeXFileFormatter())
@@ -45,20 +45,19 @@ def create_file_handler(log_file_path, level=logging.INFO, max_bytes=10*1024*102
 
 
 def get_default_log_path():
-    """Get the default log file path for SciTeX."""
-    # Use ~/.scitex/logs/ directory
-    scitex_dir = Path.home() / '.scitex'
-    logs_dir = scitex_dir / 'logs'
-    
+    """Get the default log file path for SciTeX.
+
+    Uses SCITEX_DIR environment variable with fallback to ~/.scitex.
+    Supports .env file loading for configuration.
+    """
+    scitex_dir = get_scitex_dir()
+    logs_dir = scitex_dir / "logs"
+
     # Create timestamped log file
     timestamp = datetime.now().strftime("%Y-%m-%d")
-    log_file = logs_dir / f'scitex-{timestamp}.log'
-    
+    log_file = logs_dir / f"scitex-{timestamp}.log"
+
     return str(log_file)
 
 
-__all__ = [
-    'create_console_handler',
-    'create_file_handler', 
-    'get_default_log_path'
-]
+__all__ = ["create_console_handler", "create_file_handler", "get_default_log_path"]

scitex/logging/_levels.py

@@ -6,16 +6,16 @@ import logging
 
 # Custom log levels for success/fail
 SUCCESS = 31  # Between WARNING (30) and ERROR (40)
-FAIL = 35     # Between WARNING (30) and ERROR (40)
+FAIL = 35  # Between WARNING (30) and ERROR (40)
 
 # Add custom levels to logging module with 4-character abbreviations
-logging.addLevelName(SUCCESS, 'SUCC')
-logging.addLevelName(FAIL, 'FAIL')
-logging.addLevelName(logging.DEBUG, 'DEBU')
-logging.addLevelName(logging.INFO, 'INFO')
-logging.addLevelName(logging.WARNING, 'WARN')
-logging.addLevelName(logging.ERROR, 'ERRO')
-logging.addLevelName(logging.CRITICAL, 'CRIT')
+logging.addLevelName(SUCCESS, "SUCC")
+logging.addLevelName(FAIL, "FAIL")
+logging.addLevelName(logging.DEBUG, "DEBU")
+logging.addLevelName(logging.INFO, "INFO")
+logging.addLevelName(logging.WARNING, "WARN")
+logging.addLevelName(logging.ERROR, "ERRO")
+logging.addLevelName(logging.CRITICAL, "CRIT")
 
 # Standard levels for convenience
 DEBUG = logging.DEBUG
@@ -24,12 +24,4 @@ WARNING = logging.WARNING
 ERROR = logging.ERROR
 CRITICAL = logging.CRITICAL
 
-__all__ = [
-    'SUCCESS',
-    'FAIL',
-    'DEBUG',
-    'INFO',
-    'WARNING', 
-    'ERROR',
-    'CRITICAL'
-]
+__all__ = ["SUCCESS", "FAIL", "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]

scitex/logging/_logger.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/logging/_logger.py"
-)
+
+__FILE__ = "./src/scitex/logging/_logger.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -56,21 +55,25 @@ class SciTeXLogger(logging.Logger):
             if not isinstance(message, str):
                 # Convert DotDict to regular dict for better pprint support
                 # Exclude private keys (starting with _) by default
-                if hasattr(message, 'to_dict'):
+                if hasattr(message, "to_dict"):
                     message = message.to_dict(include_private=False)
                 message = _pprint.pformat(message, indent=2, width=80, compact=False)
 
             # For multi-line messages, indent all lines after the first
             # to align with the log level prefix (e.g., "INFO: ") plus any indent
-            if '\n' in message:
-                lines = message.split('\n')
+            if "\n" in message:
+                lines = message.split("\n")
                 # Calculate the indent needed:
                 # - 6 chars for "INFO: " or "ERRO: " prefix
                 # - Plus the indent parameter for additional spacing
                 total_indent = 6 + indent
-                prefix_indent = ' ' * total_indent
+                prefix_indent = " " * total_indent
                 # Join lines with proper indentation
-                message = lines[0] + '\n' + '\n'.join(prefix_indent + line for line in lines[1:])
+                message = (
+                    lines[0]
+                    + "\n"
+                    + "\n".join(prefix_indent + line for line in lines[1:])
+                )
 
         # Add separator lines if requested
         if sep is not None:
@@ -88,7 +91,15 @@ class SciTeXLogger(logging.Logger):
         self._log(level, message, args, **kwargs)
 
     def debug(
-        self, message, *args, indent=0, sep=None, n_sep=40, c=None, pprint=False, **kwargs
+        self,
+        message,
+        *args,
+        indent=0,
+        sep=None,
+        n_sep=40,
+        c=None,
+        pprint=False,
+        **kwargs,
     ):
         """Log a debug message with optional indent, separator, color, and pprint."""
         if self.isEnabledFor(logging.DEBUG):
@@ -97,7 +108,15 @@ class SciTeXLogger(logging.Logger):
             )
 
     def info(
-        self, message, *args, indent=0, sep=None, n_sep=40, c=None, pprint=False, **kwargs
+        self,
+        message,
+        *args,
+        indent=0,
+        sep=None,
+        n_sep=40,
+        c=None,
+        pprint=False,
+        **kwargs,
     ):
         """Log an info message with optional indent, separator, color, and pprint."""
         if self.isEnabledFor(logging.INFO):
@@ -106,7 +125,15 @@ class SciTeXLogger(logging.Logger):
             )
 
     def warning(
-        self, message, *args, indent=0, sep=None, n_sep=40, c=None, pprint=False, **kwargs
+        self,
+        message,
+        *args,
+        indent=0,
+        sep=None,
+        n_sep=40,
+        c=None,
+        pprint=False,
+        **kwargs,
     ):
         """Log a warning message with optional indent, separator, color, and pprint."""
         if self.isEnabledFor(logging.WARNING):
@@ -123,7 +150,15 @@ class SciTeXLogger(logging.Logger):
             )
 
     def error(
-        self, message, *args, indent=0, sep=None, n_sep=40, c=None, pprint=False, **kwargs
+        self,
+        message,
+        *args,
+        indent=0,
+        sep=None,
+        n_sep=40,
+        c=None,
+        pprint=False,
+        **kwargs,
     ):
         """Log an error message with optional indent, separator, color, and pprint."""
         if self.isEnabledFor(logging.ERROR):
@@ -132,7 +167,15 @@ class SciTeXLogger(logging.Logger):
             )
 
     def critical(
-        self, message, *args, indent=0, sep=None, n_sep=40, c=None, pprint=False, **kwargs
+        self,
+        message,
+        *args,
+        indent=0,
+        sep=None,
+        n_sep=40,
+        c=None,
+        pprint=False,
+        **kwargs,
     ):
         """Log a critical message with optional indent, separator, color, and pprint."""
         if self.isEnabledFor(logging.CRITICAL):
@@ -149,7 +192,15 @@ class SciTeXLogger(logging.Logger):
             )
 
     def success(
-        self, message, *args, indent=0, sep=None, n_sep=40, c=None, pprint=False, **kwargs
+        self,
+        message,
+        *args,
+        indent=0,
+        sep=None,
+        n_sep=40,
+        c=None,
+        pprint=False,
+        **kwargs,
     ):
         """Log a success message with optional indent, separator, color, and pprint."""
         if self.isEnabledFor(SUCCESS):
@@ -158,7 +209,15 @@ class SciTeXLogger(logging.Logger):
             )
 
     def fail(
-        self, message, *args, indent=0, sep=None, n_sep=40, c=None, pprint=False, **kwargs
+        self,
+        message,
+        *args,
+        indent=0,
+        sep=None,
+        n_sep=40,
+        c=None,
+        pprint=False,
+        **kwargs,
     ):
         """Log a failure message with optional indent, separator, color, and pprint."""
         if self.isEnabledFor(FAIL):

scitex/logging/_print_capture.py

@@ -10,53 +10,53 @@ from io import StringIO
 
 class PrintCapture:
     """Capture print() output and redirect to logging system."""
-    
+
     def __init__(self, logger_name: str = "scitex.print_capture"):
         self.logger = logging.getLogger(logger_name)
         self.original_stdout = sys.stdout
         self.original_stderr = sys.stderr
         self.capturing = False
-        
+
     def start_capture(self):
         """Start capturing print output."""
         if self.capturing:
             return
-            
+
         self.capturing = True
         sys.stdout = self
-        
+
     def stop_capture(self):
         """Stop capturing print output."""
         if not self.capturing:
             return
-            
+
         self.capturing = False
         sys.stdout = self.original_stdout
-        
+
     def write(self, text):
         """Handle write calls from print()."""
         # Also write to original stdout for real-time viewing
         self.original_stdout.write(text)
         self.original_stdout.flush()
-        
+
         # Log the output (strip newlines for cleaner logs)
-        clean_text = text.rstrip('\n')
+        clean_text = text.rstrip("\n")
         if clean_text:  # Only log non-empty text
             self.logger.info(clean_text)
-            
+
     def flush(self):
         """Handle flush calls."""
         self.original_stdout.flush()
-        
+
     def isatty(self):
         """Handle isatty calls."""
         return self.original_stdout.isatty()
-        
+
     def __enter__(self):
         """Context manager entry."""
         self.start_capture()
         return self
-        
+
     def __exit__(self, exc_type, exc_val, exc_tb):
         """Context manager exit."""
         self.stop_capture()
@@ -88,8 +88,8 @@ def is_print_capture_enabled():
 
 
 __all__ = [
-    'PrintCapture',
-    'enable_print_capture', 
-    'disable_print_capture',
-    'is_print_capture_enabled'
-]
+    "PrintCapture",
+    "enable_print_capture",
+    "disable_print_capture",
+    "is_print_capture_enabled",
+]

scitex/nn/_BNet.py

@@ -39,9 +39,7 @@ class BNet(nn.Module):
 
         self.sc = SwapChannels()
         self.dc = DropoutChannels(dropout=0.01)
-        self.fgc = FreqGainChanger(
-            BNet_config["n_bands"], BNet_config["SAMP_RATE"]
-        )
+        self.fgc = FreqGainChanger(BNet_config["n_bands"], BNet_config["SAMP_RATE"])
         self.heads = nn.ModuleList(
             [
                 BHead(n_ch, N_VIRTUAL_CHS).to(self.dummy_param.device)

scitex/nn/_Filters.py

@@ -85,7 +85,9 @@ class BaseFilter1D(nn.Module):
             n_chs,
             len(self.kernels),
             seq_len,
-        ), f"The shape of the filtered signal ({x.shape}) does not match the expected shape: ({batch_size}, {n_chs}, {len(self.kernels)}, {seq_len})."
+        ), (
+            f"The shape of the filtered signal ({x.shape}) does not match the expected shape: ({batch_size}, {n_chs}, {len(self.kernels)}, {seq_len})."
+        )
 
         # Edge remove
         x = self.remove_edges(x, edge_len)
@@ -411,7 +413,9 @@ if __name__ == "__main__":
     import scitex
 
     # Start
-    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(sys, plt, fig_scale=5)
+    CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(
+        sys, plt, fig_scale=5
+    )
 
     xx, tt, fs = scitex.dsp.demo_sig(sig_type="chirp", fs=1024)
     xx = torch.tensor(xx).cuda()

scitex/nn/_ModulationIndex.py

@@ -170,7 +170,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/nn/_PAC.py

@@ -212,7 +212,6 @@ class PAC(nn.Module):
         trainable=False,
         fp16=False,
     ):
-
         # A static, gen purpose BandPassFilter
         if not trainable:
             # First, bands definitions for phase and amplitude are declared
@@ -361,7 +360,9 @@ if __name__ == "__main__":
     plt.bar(xx, amp_prob[0])
     """
 
-    scitex.gen.print_block(f"PAC calculation time: {ts.delta(-1,-2):.3f} sec", c="yellow")
+    scitex.gen.print_block(
+        f"PAC calculation time: {ts.delta(-1, -2):.3f} sec", c="yellow"
+    )
     # 0.17 sec
     scitex.gen.print_block(
         f"x.shape: {xx.shape}"

scitex/nn/_PSD.py

@@ -14,7 +14,6 @@ class PSD(nn.Module):
         self.prob = prob
 
     def forward(self, signal):
-
         is_complex = signal.is_complex()
         if is_complex:
             signal_fft = torch.fft.fft(signal, dim=self.dim)