steer-core 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

steer_core/Apps/ContextManagers.py

@@ -5,10 +5,14 @@ from typing import List, Dict
 
 
 @contextmanager
-def capture_warnings(existing_warnings: List[Dict] = None, source: str = None, clear_source_warnings: bool = True):
+def capture_warnings(
+    existing_warnings: List[Dict] = None,
+    source: str = None,
+    clear_source_warnings: bool = True,
+):
     """
     Context manager to capture warnings and format them for Dash.
-    
+
     Parameters
     ----------
     existing_warnings : List[Dict], optional
@@ -17,7 +21,7 @@ def capture_warnings(existing_warnings: List[Dict] = None, source: str = None, c
         Source identifier for the warnings
     clear_source_warnings : bool, optional
         Whether to clear existing warnings from the same source before adding new ones
-        
+
     Yields
     ------
     List[Dict]
@@ -25,31 +29,33 @@ def capture_warnings(existing_warnings: List[Dict] = None, source: str = None, c
     """
     if existing_warnings is None:
         existing_warnings = []
-    
+
     # Clear existing warnings from the same source if requested
     if clear_source_warnings and source:
-        existing_warnings = [w for w in existing_warnings if w.get('source') != source]
-    
+        existing_warnings = [w for w in existing_warnings if w.get("source") != source]
+
     with warnings.catch_warnings(record=True) as warnings_list:
         warnings.simplefilter("always")
-        
+
         # Yield the cleaned warnings list
         yield existing_warnings
-        
+
         # Process any new warnings after the operation
         new_warnings = []
         for w in warnings_list:
-            
-            new_warnings.append({
-                'message': str(w.message),
-                'category': w.category.__name__,
-                'filename': w.filename,
-                'lineno': w.lineno,
-                'timestamp': time.time(),
-                'source': source or 'unknown',
-                'id': f"{source}_{int(time.time())}" if source else f"warning_{int(time.time())}"
-            })
-        
+            new_warnings.append(
+                {
+                    "message": str(w.message),
+                    "category": w.category.__name__,
+                    "filename": w.filename,
+                    "lineno": w.lineno,
+                    "timestamp": time.time(),
+                    "source": source or "unknown",
+                    "id": f"{source}_{int(time.time())}"
+                    if source
+                    else f"warning_{int(time.time())}",
+                }
+            )
+
         # Add new warnings
         existing_warnings.extend(new_warnings)
-

steer_core/Apps/Performance/CallbackTimer.py

@@ -4,6 +4,7 @@ import time
 
 def timed_callback(func):
     """Time callback execution for optimization."""
+
     @wraps(func)
     def wrapper(*args, **kwargs):
         start = time.time()
@@ -11,7 +12,7 @@ def timed_callback(func):
         duration = time.time() - start
 
         print(f"CALLBACK: {func.__name__} took {duration:.2f}s")
-    
+
         return result
-    return wrapper
 
+    return wrapper