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

scitex/utils/_search.py

@@ -71,17 +71,17 @@ def search(
         # Check for numpy arrays first
         if isinstance(string_or_pattern, np.ndarray):
             return string_or_pattern.tolist()
-        
+
         # Check for pandas types if pandas is available
         if pd is not None:
             if isinstance(string_or_pattern, (pd.Series, pd.Index)):
                 return string_or_pattern.tolist()
-        
+
         # Check for xarray types if xarray is available
         if xr is not None:
             if isinstance(string_or_pattern, xr.DataArray):
                 return string_or_pattern.tolist()
-        
+
         # Check for other iterables
         if isinstance(string_or_pattern, abc.KeysView):
             return list(string_or_pattern)
@@ -106,9 +106,9 @@ def search(
     keys_matched = list(np.array(strings)[indices_matched])
 
     if ensure_one:
-        assert (
-            len(indices_matched) == 1
-        ), "Expected exactly one match, but found {}".format(len(indices_matched))
+        assert len(indices_matched) == 1, (
+            "Expected exactly one match, but found {}".format(len(indices_matched))
+        )
 
     if as_bool:
         bool_matched = np.zeros(len(strings), dtype=bool)

scitex/utils/_verify_scitex_format.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/utils/_verify_scitex_format.py"
-)
+
+__FILE__ = "./src/scitex/utils/_verify_scitex_format.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -48,6 +47,8 @@ logger = logging.getLogger(__name__)
 
 
 """Functions & Classes"""
+
+
 @dataclass
 class FileInfo:
     """Store file information."""
@@ -104,9 +105,7 @@ class TemplateCompliance:
         return sum(checks) / len(checks)
 
 
-def scan_python_files(
-    paths: list[Path], base_dir: Path = None
-) -> Dict[str, FileInfo]:
+def scan_python_files(paths: list[Path], base_dir: Path = None) -> Dict[str, FileInfo]:
     """Scan Python files from filesystem paths.
 
     Args:
@@ -136,9 +135,7 @@ def scan_python_files(
 
         for py_file in python_files:
             # Skip template.py, __init__.py, and test files
-            if any(
-                skip in py_file.name for skip in ["template.py", "__init__.py"]
-            ):
+            if any(skip in py_file.name for skip in ["template.py", "__init__.py"]):
                 continue
 
             try:
@@ -162,9 +159,7 @@ def check_compliance(content: str) -> TemplateCompliance:
     compliance = TemplateCompliance()
 
     # Check for main function
-    compliance.has_main = bool(
-        re.search(r"^def main\(", content, re.MULTILINE)
-    )
+    compliance.has_main = bool(re.search(r"^def main\(", content, re.MULTILINE))
 
     # Check for parse_args function
     compliance.has_parse_args = bool(
@@ -172,14 +167,10 @@ def check_compliance(content: str) -> TemplateCompliance:
     )
 
     # Check for run_main function
-    compliance.has_run_main = bool(
-        re.search(r"^def run_main\(", content, re.MULTILINE)
-    )
+    compliance.has_run_main = bool(re.search(r"^def run_main\(", content, re.MULTILINE))
 
     # Check for main guard
-    compliance.has_main_guard = bool(
-        re.search(r'if __name__ == "__main__":', content)
-    )
+    compliance.has_main_guard = bool(re.search(r'if __name__ == "__main__":', content))
 
     # Check if run_main follows the exact template format
     compliance.is_run_main_unchanged = _check_run_main_unchanged(content)
@@ -194,9 +185,7 @@ def check_compliance(content: str) -> TemplateCompliance:
     )
 
     # Check for scitex session usage
-    compliance.uses_scitex_session = bool(
-        re.search(r"stx\.session\.start", content)
-    )
+    compliance.uses_scitex_session = bool(re.search(r"stx\.session\.start", content))
 
     # Check for verbose parameter
     compliance.has_verbose_param = bool(
@@ -250,7 +239,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
     lines.append(f"Total Python files analyzed: {total_files}")
     if total_files > 0:
         lines.append(
-            f"Fully compliant files: {compliant_files} ({compliant_files/total_files*100:.1f}%)"
+            f"Fully compliant files: {compliant_files} ({compliant_files / total_files * 100:.1f}%)"
         )
         lines.append(f"Average compliance score: {avg_score:.1%}")
     else:
@@ -280,9 +269,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
             status = "✗ NON-COMPLIANT"
 
         lines.append(f"{status} [{score:.0%}] {filepath}")
-        lines.append(
-            f"  Lines: {file_info.lines}, Size: {file_info.size} bytes"
-        )
+        lines.append(f"  Lines: {file_info.lines}, Size: {file_info.size} bytes")
 
         # Show missing components
         if not compliance.is_compliant:
@@ -315,9 +302,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
             optional_missing.append("verbose parameter")
 
         if optional_missing:
-            lines.append(
-                f"  ⚠️  OPTIONAL Missing: {', '.join(optional_missing)}"
-            )
+            lines.append(f"  ⚠️  OPTIONAL Missing: {', '.join(optional_missing)}")
 
         lines.append("")
 
@@ -358,11 +343,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
         ),
         (
             "  'Functions & Classes'",
-            sum(
-                1
-                for _, c in results.values()
-                if c.has_functions_classes_section
-            ),
+            sum(1 for _, c in results.values() if c.has_functions_classes_section),
         ),
         (
             "  scitex session",
@@ -411,9 +392,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
             if not compliance.has_main_guard:
                 fixes.append("  - Add if __name__ == '__main__': run_main()")
             if not compliance.is_run_main_unchanged:
-                fixes.append(
-                    "  - Update run_main() to match template format exactly"
-                )
+                fixes.append("  - Update run_main() to match template format exactly")
             for fix in fixes:
                 lines.append(fix)
             lines.append("")
@@ -441,9 +420,7 @@ def main(args):
     else:
         # Default to current directory
         paths = [Path.cwd()]
-        base_dir = (
-            Path(args.base_dir).resolve() if args.base_dir else Path.cwd()
-        )
+        base_dir = Path(args.base_dir).resolve() if args.base_dir else Path.cwd()
 
     logger.info(f"Scanning paths: {[str(p) for p in paths]}")
     logger.info(f"Base directory: {base_dir}")
@@ -479,9 +456,7 @@ def main(args):
         logger.success("All files are compliant!")
         return 0
     else:
-        logger.warning(
-            f"{len(results) - compliant_count} files need attention"
-        )
+        logger.warning(f"{len(results) - compliant_count} files need attention")
         return 0  # Still return 0 to avoid breaking pipelines
 
 

scitex/utils/_verify_scitex_format_v01.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/utils/_verify_scitex_format.py"
-)
+
+__FILE__ = "./src/scitex/utils/_verify_scitex_format.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 """Analyze template compliance from document text."""
@@ -94,10 +93,7 @@ def parse_document(doc_text: str) -> Dict[str, FileInfo]:
         content = match.group(5)
 
         # Only process .py files, skip template.py
-        if (
-            relative_path.endswith(".py")
-            and "template.py" not in relative_path
-        ):
+        if relative_path.endswith(".py") and "template.py" not in relative_path:
             files[relative_path] = FileInfo(
                 relative_path=relative_path,
                 content=content,
@@ -113,9 +109,7 @@ def check_compliance(content: str) -> TemplateCompliance:
     compliance = TemplateCompliance()
 
     # Check for main function
-    compliance.has_main = bool(
-        re.search(r"^def main\(", content, re.MULTILINE)
-    )
+    compliance.has_main = bool(re.search(r"^def main\(", content, re.MULTILINE))
 
     # Check for parse_args function
     compliance.has_parse_args = bool(
@@ -123,14 +117,10 @@ def check_compliance(content: str) -> TemplateCompliance:
     )
 
     # Check for run_main function
-    compliance.has_run_main = bool(
-        re.search(r"^def run_main\(", content, re.MULTILINE)
-    )
+    compliance.has_run_main = bool(re.search(r"^def run_main\(", content, re.MULTILINE))
 
     # Check for main guard
-    compliance.has_main_guard = bool(
-        re.search(r'if __name__ == "__main__":', content)
-    )
+    compliance.has_main_guard = bool(re.search(r'if __name__ == "__main__":', content))
 
     # Check for module docstring
     compliance.has_docstring = bool(re.search(r'"""[\s\S]*?"""', content))
@@ -142,9 +132,7 @@ def check_compliance(content: str) -> TemplateCompliance:
     )
 
     # Check for scitex session usage
-    compliance.uses_scitex_session = bool(
-        re.search(r"stx\.session\.start", content)
-    )
+    compliance.uses_scitex_session = bool(re.search(r"stx\.session\.start", content))
 
     # Check for verbose parameter
     compliance.has_verbose_param = bool(
@@ -173,7 +161,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
 
     lines.append(f"Total Python files analyzed: {total_files}")
     lines.append(
-        f"Fully compliant files: {compliant_files} ({compliant_files/total_files*100:.1f}%)"
+        f"Fully compliant files: {compliant_files} ({compliant_files / total_files * 100:.1f}%)"
     )
     lines.append(f"Average compliance score: {avg_score:.1%}")
     lines.append("")
@@ -201,9 +189,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
             status = "✗ NON-COMPLIANT"
 
         lines.append(f"{status} [{score:.0%}] {filepath}")
-        lines.append(
-            f"  Lines: {file_info.lines}, Size: {file_info.size} bytes"
-        )
+        lines.append(f"  Lines: {file_info.lines}, Size: {file_info.size} bytes")
 
         # Show missing components
         if not compliance.is_compliant:
@@ -234,9 +220,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
             optional_missing.append("verbose parameter")
 
         if optional_missing:
-            lines.append(
-                f"  ⚠️  OPTIONAL Missing: {', '.join(optional_missing)}"
-            )
+            lines.append(f"  ⚠️  OPTIONAL Missing: {', '.join(optional_missing)}")
 
         lines.append("")
 
@@ -273,11 +257,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
         ),
         (
             "  'Functions & Classes'",
-            sum(
-                1
-                for _, c in results.values()
-                if c.has_functions_classes_section
-            ),
+            sum(1 for _, c in results.values() if c.has_functions_classes_section),
         ),
         (
             "  scitex session",
@@ -337,9 +317,7 @@ def generate_report(results: Dict[str, tuple]) -> str:
 def main():
     """Main function."""
     # Read the document
-    with open(
-        "/mnt/user-data/uploads/document_1.txt", "r", encoding="utf-8"
-    ) as f:
+    with open("/mnt/user-data/uploads/document_1.txt", "r", encoding="utf-8") as f:
         doc_text = f.read()
 
     print("Parsing document...")

scitex/utils/template.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/utils/template.py"
-)
+
+__FILE__ = "./src/scitex/utils/template.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -55,6 +54,8 @@ logger = logging.getLogger(__name__)
 # CONFIG = stx.io.load_configs()
 
 """Functions & Classes"""
+
+
 def main(args):
     return 0
 

scitex/vis/backend/__init__.py

@@ -8,7 +8,7 @@ This module bridges the gap between JSON specifications and
 actual matplotlib figures using scitex.plt.
 """
 
-from .parser import (
+from ._parser import (
     parse_figure_json,
     parse_axes_json,
     parse_plot_json,
@@ -17,7 +17,7 @@ from .parser import (
     validate_figure_json,
 )
 
-from .render import (
+from ._render import (
     render_figure,
     render_axes,
     render_plot,
@@ -26,7 +26,7 @@ from .render import (
     build_figure_from_json,
 )
 
-from .export import (
+from ._export import (
     export_figure,
     export_figure_from_file,
     export_multiple_formats,

scitex/vis/backend/{export.py → _export.py}

@@ -6,7 +6,7 @@
 from pathlib import Path
 from typing import Dict, Any, Optional, Union
 
-from .render import build_figure_from_json
+from ._render import build_figure_from_json
 
 
 def export_figure(

scitex/vis/backend/{parser.py → _parser.py}

@@ -69,9 +69,7 @@ def parse_axes_json(axes_json: Dict[str, Any]) -> AxesModel:
 
     # Parse nested annotations
     annotations_data = axes_json.get("annotations", [])
-    parsed_annotations = [
-        parse_annotation_json(ann) for ann in annotations_data
-    ]
+    parsed_annotations = [parse_annotation_json(ann) for ann in annotations_data]
 
     # Parse nested guides
     guides_data = axes_json.get("guides", [])

scitex/vis/backend/{render.py → _render.py}

@@ -7,7 +7,7 @@ from typing import Dict, Any, Optional
 import numpy as np
 
 from ..model import FigureModel, AxesModel, PlotModel, GuideModel, AnnotationModel
-from .parser import parse_figure_json
+from ._parser import parse_figure_json
 
 
 def render_figure(fig_model: FigureModel):

scitex/vis/canvas.py

@@ -60,7 +60,12 @@ class Canvas:
             "author": "",
             "description": "",
         }
-        self._caption: Dict[str, Any] = {"text": "", "render": False, "fontsize": 10, "width_mm": None}
+        self._caption: Dict[str, Any] = {
+            "text": "",
+            "render": False,
+            "fontsize": 10,
+            "width_mm": None,
+        }
         self._source_files: Dict[str, Path] = {}  # panel_name -> source_path
 
     @property
@@ -148,7 +153,11 @@ class Canvas:
 
         # Apply any additional kwargs to nested dicts
         for key, value in kwargs.items():
-            if key in panel_entry and isinstance(panel_entry[key], dict) and isinstance(value, dict):
+            if (
+                key in panel_entry
+                and isinstance(panel_entry[key], dict)
+                and isinstance(value, dict)
+            ):
                 panel_entry[key].update(value)
             elif key not in ["rotation_deg", "opacity", "flip_h", "flip_v", "visible"]:
                 panel_entry[key] = value
@@ -287,7 +296,9 @@ class Canvas:
 
         return self
 
-    def set_title(self, text: str, position: tuple = None, fontsize: int = 14) -> "Canvas":
+    def set_title(
+        self, text: str, position: tuple = None, fontsize: int = 14
+    ) -> "Canvas":
         """
         Set canvas title.
 
@@ -405,6 +416,7 @@ class Canvas:
         >>> canvas.save("/output/fig1.canvas", bundle=True)  # Copies files
         """
         import scitex as stx
+
         return stx.io.save(self, path, bundle=bundle, **kwargs)
 
     def __repr__(self) -> str: