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

scitex/dt/_normalize_timestamp.py

@@ -5,6 +5,7 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
+
 __FILE__ = __file__
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
@@ -44,7 +45,7 @@ import scitex as stx
 CONFIG = stx.io.load_configs()
 
 # Get standard format from config
-STANDARD_FORMAT = CONFIG.FORMATS.TIMESTAMP or ... fixme
+STANDARD_FORMAT = CONFIG.FORMATS.TIMESTAMP or "%Y-%m-%d %H:%M:%S"
 
 
 # Common alternative formats to try when parsing
@@ -67,6 +68,8 @@ ALTERNATIVE_FORMATS = [
 ]
 
 """Functions & Classes"""
+
+
 def normalize_timestamp(
     timestamp: Union[datetime, str, int, float],
     return_as: str = "str",

scitex/errors.py

@@ -5,6 +5,7 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
+
 __FILE__ = __file__
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
@@ -305,9 +306,7 @@ class DataError(SciTeXError):
 class ShapeError(DataError):
     """Raised when data shapes are incompatible."""
 
-    def __init__(
-        self, expected_shape: tuple, actual_shape: tuple, operation: str
-    ):
+    def __init__(self, expected_shape: tuple, actual_shape: tuple, operation: str):
         super().__init__(
             f"Shape mismatch in {operation}",
             context={
@@ -492,9 +491,7 @@ def check_file_exists(filepath: str) -> None:
         raise PathNotFoundError(filepath)
 
 
-def check_shape_compatibility(
-    shape1: tuple, shape2: tuple, operation: str
-) -> None:
+def check_shape_compatibility(shape1: tuple, shape2: tuple, operation: str) -> None:
     """Check if two shapes are compatible for an operation.
 
     Parameters

scitex/etc/__init__.py

@@ -12,4 +12,4 @@ such as keyboard input handling for interactive programs.
 
 from .wait_key import wait_key, count
 
-__all__ = ["wait_key", "count"]
+__all__ = ["wait_key", "count"]

scitex/gen/_DimHandler.py

@@ -78,9 +78,9 @@ class DimHandler:
         Returns:
             x_flattened (numpy.ndarray): The reshaped array with kept dimensions moved to the end.
         """
-        assert len(keepdims) <= len(
-            x.shape
-        ), "keepdims cannot have more dimensions than the array itself."
+        assert len(keepdims) <= len(x.shape), (
+            "keepdims cannot have more dimensions than the array itself."
+        )
 
         # Normalize negative indices to positive indices
         total_dims = len(x.shape)
@@ -117,9 +117,9 @@ class DimHandler:
         Note:
             This method modifies the `orig_shape`, `keepdims`, `n_non_keepdims`, and `n_keepdims` attributes based on the input.
         """
-        assert len(keepdims) <= len(
-            x.shape
-        ), "keepdims cannot have more dimensions than the tensor itself."
+        assert len(keepdims) <= len(x.shape), (
+            "keepdims cannot have more dimensions than the tensor itself."
+        )
 
         keepdims = torch.tensor(keepdims).clone().detach().cpu().int()
         # Normalize negative indices to positive indices

scitex/gen/__init__.py

@@ -7,6 +7,7 @@ from ._alternate_kwarg import alternate_kwarg
 from ._cache import cache
 from ._check_host import check_host, is_host, verify_host
 from ._ci import ci
+
 # _close.py moved to old/ directory - functionality now in scitex.session
 from ._embed import embed
 from ._inspect_module import inspect_module
@@ -31,7 +32,10 @@ from ._src import src
 
 # BACKWARD COMPATIBILITY: Import deprecated wrappers
 from ._deprecated_start import start as _deprecated_start
-from ._deprecated_close import close as _deprecated_close, running2finished as _deprecated_running2finished
+from ._deprecated_close import (
+    close as _deprecated_close,
+    running2finished as _deprecated_running2finished,
+)
 
 # Override the imported functions with deprecated wrappers
 start = _deprecated_start

scitex/gen/_deprecated_close.py

@@ -5,6 +5,7 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
+
 __FILE__ = __file__
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------

scitex/gen/_deprecated_start.py

@@ -14,13 +14,15 @@ from scitex.decorators._deprecated import deprecated
 
 @deprecated(
     reason="Use scitex.session.start instead. The old interface will be removed in a future version.",
-    forward_to="..session.start"
+    forward_to="..session.start",
 )
 def start(*args, **kwargs):
     """This docstring will be auto-generated by the @deprecated decorator."""
     # This function body will never be executed due to forwarding
     # but we need a placeholder implementation
-    raise NotImplementedError("This function should be forwarded by the @deprecated decorator")
+    raise NotImplementedError(
+        "This function should be forwarded by the @deprecated decorator"
+    )
 
 
-__all__ = ['start']
+__all__ = ["start"]

scitex/gen/_detect_environment.py

@@ -17,20 +17,20 @@ import os
 import sys
 from typing import Literal, Tuple
 
-__all__ = ['detect_environment', 'get_output_directory']
+__all__ = ["detect_environment", "get_output_directory"]
 
-EnvironmentType = Literal['script', 'jupyter', 'ipython', 'interactive', 'unknown']
+EnvironmentType = Literal["script", "jupyter", "ipython", "interactive", "unknown"]
 
 
 def detect_environment() -> EnvironmentType:
     """
     Detect the current execution environment.
-    
+
     Returns
     -------
     EnvironmentType
         One of: 'script', 'jupyter', 'ipython', 'interactive', 'unknown'
-        
+
     Examples
     --------
     >>> env = detect_environment()
@@ -38,49 +38,51 @@ def detect_environment() -> EnvironmentType:
     Running in: script
     """
     # Check for Jupyter notebook
-    if 'ipykernel' in sys.modules:
+    if "ipykernel" in sys.modules:
         try:
             ip = get_ipython()
-            if ip and type(ip).__name__ == 'ZMQInteractiveShell':
-                return 'jupyter'
+            if ip and type(ip).__name__ == "ZMQInteractiveShell":
+                return "jupyter"
         except NameError:
             pass
-    
+
     # Check for IPython console
     try:
         ip = get_ipython()
-        if ip and type(ip).__name__ == 'TerminalInteractiveShell':
-            return 'ipython'
+        if ip and type(ip).__name__ == "TerminalInteractiveShell":
+            return "ipython"
     except NameError:
         pass
-    
+
     # Check for regular Python script
-    if sys.argv and sys.argv[0] and sys.argv[0].endswith('.py'):
-        return 'script'
-    
+    if sys.argv and sys.argv[0] and sys.argv[0].endswith(".py"):
+        return "script"
+
     # Check for interactive Python
-    if hasattr(sys, 'ps1'):
-        return 'interactive'
-    
-    return 'unknown'
+    if hasattr(sys, "ps1"):
+        return "interactive"
+
+    return "unknown"
 
 
-def get_output_directory(specified_path: str, env_type: EnvironmentType = None) -> Tuple[str, bool]:
+def get_output_directory(
+    specified_path: str, env_type: EnvironmentType = None
+) -> Tuple[str, bool]:
     """
     Get the appropriate output directory based on environment.
-    
+
     Parameters
     ----------
     specified_path : str
         The path specified by the user
     env_type : EnvironmentType, optional
         Override environment detection
-        
+
     Returns
     -------
     tuple[str, bool]
         (output_directory, should_use_temp)
-        
+
     Examples
     --------
     >>> output_dir, use_temp = get_output_directory("data.csv")
@@ -88,56 +90,56 @@ def get_output_directory(specified_path: str, env_type: EnvironmentType = None)
     Save to: ./script_out/data.csv, Temp: False
     """
     import inspect
-    
+
     if env_type is None:
         env_type = detect_environment()
-    
+
     # For absolute paths, use as-is
-    if specified_path.startswith('/'):
+    if specified_path.startswith("/"):
         return specified_path, False
-    
+
     # Get base directory based on environment
-    if env_type == 'script':
+    if env_type == "script":
         # Use script location
         try:
             script_path = inspect.stack()[1].filename
-            if script_path and not script_path.startswith('<'):
+            if script_path and not script_path.startswith("<"):
                 script_dir = os.path.dirname(os.path.abspath(script_path))
                 script_name = os.path.splitext(os.path.basename(script_path))[0]
                 base_dir = os.path.join(script_dir, f"{script_name}_out")
                 return os.path.join(base_dir, specified_path), False
         except:
             pass
-    
-    elif env_type == 'jupyter':
+
+    elif env_type == "jupyter":
         # For Jupyter, use current working directory with subdirectory
         # This keeps outputs near the notebook
-        base_dir = os.path.join(os.getcwd(), 'notebook_outputs')
+        base_dir = os.path.join(os.getcwd(), "notebook_outputs")
         return os.path.join(base_dir, specified_path), False
-    
-    elif env_type in ['ipython', 'interactive']:
+
+    elif env_type in ["ipython", "interactive"]:
         # Use temp directory for console sessions
-        user = os.getenv('USER', 'unknown')
-        base_dir = f'/tmp/{user}/{env_type}'
+        user = os.getenv("USER", "unknown")
+        base_dir = f"/tmp/{user}/{env_type}"
         return os.path.join(base_dir, specified_path), True
-    
+
     # Fallback: use current directory
-    return os.path.join('./output', specified_path), False
+    return os.path.join("./output", specified_path), False
 
 
 def is_notebook() -> bool:
     """Check if running in Jupyter notebook."""
-    return detect_environment() == 'jupyter'
+    return detect_environment() == "jupyter"
 
 
 def is_ipython() -> bool:
     """Check if running in IPython (console or notebook)."""
-    return detect_environment() in ['jupyter', 'ipython']
+    return detect_environment() in ["jupyter", "ipython"]
 
 
 def is_script() -> bool:
     """Check if running as a script."""
-    return detect_environment() == 'script'
+    return detect_environment() == "script"
 
 
 # Backward compatibility
@@ -149,4 +151,5 @@ def is_ipython_legacy() -> bool:
     except NameError:
         return False
 
-# EOF
+
+# EOF

scitex/gen/_detect_notebook_path.py

@@ -14,18 +14,18 @@ import os
 import json
 from typing import Optional
 
-__all__ = ['get_notebook_path', 'get_notebook_output_dir']
+__all__ = ["get_notebook_path", "get_notebook_output_dir"]
 
 
 def get_notebook_path() -> Optional[str]:
     """
     Get the path of the currently running Jupyter notebook.
-    
+
     Returns
     -------
     Optional[str]
         Path to the notebook file, or None if not in a notebook
-        
+
     Examples
     --------
     >>> path = get_notebook_path()
@@ -35,135 +35,139 @@ def get_notebook_path() -> Optional[str]:
     try:
         # Method 1: Try to get from IPython
         ip = get_ipython()
-        
+
         # Check if we're in a notebook
-        if not (ip and type(ip).__name__ == 'ZMQInteractiveShell'):
+        if not (ip and type(ip).__name__ == "ZMQInteractiveShell"):
             return None
-            
+
         # Method 2: Check IPython's notebook_name
-        if hasattr(ip, 'notebook_name'):
+        if hasattr(ip, "notebook_name"):
             return ip.notebook_name
-            
+
         # Method 3: Try to get from kernel connection file
         import re
         import glob
-        
+
         # Get kernel ID from the current session
         kernel_id = re.search(
-            r'kernel-(.*?)\.json',
-            ip.config['IPKernelApp']['connection_file']
+            r"kernel-(.*?)\.json", ip.config["IPKernelApp"]["connection_file"]
         ).group(1)
-        
+
         # Search for notebook sessions
-        runtime_dir = os.environ.get('JUPYTER_RUNTIME_DIR', 
-                                     os.path.expanduser('~/.local/share/jupyter/runtime'))
-        
-        for nbserver in glob.glob(os.path.join(runtime_dir, 'nbserver-*.json')):
+        runtime_dir = os.environ.get(
+            "JUPYTER_RUNTIME_DIR", os.path.expanduser("~/.local/share/jupyter/runtime")
+        )
+
+        for nbserver in glob.glob(os.path.join(runtime_dir, "nbserver-*.json")):
             try:
-                with open(nbserver, 'r') as f:
+                with open(nbserver, "r") as f:
                     server_info = json.load(f)
-                    
+
                 # Check notebook sessions
                 import requests
+
                 sessions_url = f"{server_info['url']}api/sessions?token={server_info.get('token', '')}"
                 response = requests.get(sessions_url)
-                
+
                 if response.status_code == 200:
                     sessions = response.json()
                     for session in sessions:
-                        if session['kernel']['id'] == kernel_id:
+                        if session["kernel"]["id"] == kernel_id:
                             # Found our notebook!
-                            notebook_path = session['notebook']['path']
+                            notebook_path = session["notebook"]["path"]
                             return notebook_path
             except:
                 continue
-                
+
         # Method 4: Try JavaScript bridge (if available)
         try:
             from IPython.display import Javascript, display
             import time
-            
+
             # This won't work in papermill, but works in interactive notebooks
-            display(Javascript('''
+            display(
+                Javascript("""
                 IPython.notebook.kernel.execute(
                     `__notebook_path__ = '${IPython.notebook.notebook_path}'`
                 );
-            '''))
-            
+            """)
+            )
+
             # Brief pause for JS execution
             time.sleep(0.1)
-            
-            if '__notebook_path__' in globals():
-                return globals()['__notebook_path__']
+
+            if "__notebook_path__" in globals():
+                return globals()["__notebook_path__"]
         except:
             pass
-            
+
     except Exception:
         pass
-        
+
     return None
 
 
 def get_notebook_output_dir(notebook_path: Optional[str] = None) -> Optional[str]:
     """
     Get the output directory for a notebook.
-    
+
     Parameters
     ----------
     notebook_path : Optional[str]
         Path to notebook, or None to auto-detect
-        
+
     Returns
     -------
     Optional[str]
         Output directory path like ./examples/analysis_out/
-        
+
     Examples
     --------
     >>> output_dir = get_notebook_output_dir()
     >>> print(output_dir)
     ./examples/analysis_out/
-    
+
     >>> output_dir = get_notebook_output_dir('./docs/tutorial.ipynb')
     >>> print(output_dir)
     ./docs/tutorial_out/
     """
     if notebook_path is None:
         notebook_path = get_notebook_path()
-        
+
     if not notebook_path:
         return None
-        
+
     # Get directory and base name
-    notebook_dir = os.path.dirname(notebook_path) or '.'
+    notebook_dir = os.path.dirname(notebook_path) or "."
     notebook_base = os.path.splitext(os.path.basename(notebook_path))[0]
-    
+
     # Create output directory path
     output_dir = os.path.join(notebook_dir, f"{notebook_base}_out")
-    
+
     return output_dir
 
 
 def detect_notebook_from_cwd() -> Optional[str]:
     """
     Fallback: Try to detect notebook from current working directory.
-    
+
     If there's exactly one .ipynb file in the current directory,
     assume that's the running notebook.
-    
+
     Returns
     -------
     Optional[str]
         Path to notebook or None
     """
     import glob
-    
-    notebooks = glob.glob('*.ipynb')
-    
+
+    notebooks = glob.glob("*.ipynb")
+
     # Only use this method if there's exactly one notebook
     if len(notebooks) == 1:
         return notebooks[0]
-        
+
     return None
 
-# EOF
+
+# EOF

scitex/gen/_embed.py

@@ -66,7 +66,7 @@ if __name__ == "__main__":
     import sys
     import matplotlib.pyplot as plt
     import scitex
-    
+
     # Start
     CONFIG, sys.stdout, sys.stderr, plt, CC = scitex.session.start(sys, plt)