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

scitex/io/_save.py

@@ -61,6 +61,7 @@ from ._save_modules import save_torch
 from ._save_modules import save_yaml
 from ._save_modules import save_zarr
 from ._save_modules._bibtex import save_bibtex
+from ._save_modules._canvas import save_canvas
 
 logger = logging.getLogger()
 
@@ -160,6 +161,7 @@ def save(
     auto_crop: bool = True,
     crop_margin_mm: float = 1.0,
     metadata_extra: dict = None,
+    json_schema: str = "recipe",
     **kwargs,
 ) -> None:
     """
@@ -202,6 +204,11 @@ def save(
                 }
             }
         Default is None.
+    json_schema : str, optional
+        Schema type for JSON metadata output. Options:
+        - "recipe": Minimal schema with method calls + data refs (default, ~95% smaller)
+        - "verbose": Full schema with all artist details
+        Default is "recipe".
     **kwargs
         Additional keyword arguments to pass to the underlying save function of the specific format.
 
@@ -295,9 +302,7 @@ def save(
                         elif var in frame.f_globals:
                             format_dict[var] = frame.f_globals[var]
                     else:
-                        raise ValueError(
-                            f"Invalid variable name in f-string: {var}"
-                        )
+                        raise ValueError(f"Invalid variable name in f-string: {var}")
 
                 # Use str.format() which is safe
                 specified_path = path_content.format(**format_dict)
@@ -369,7 +374,7 @@ def save(
                     or ("<stdin>" in script_path)
                     or env_type in ["ipython", "interactive"]
                 ):
-                    script_path = f'/tmp/{_os.getenv("USER")}'
+                    script_path = f"/tmp/{_os.getenv('USER')}"
                     sdir = script_path
                 else:
                     # Unknown environment, use current directory
@@ -426,6 +431,7 @@ def save(
             auto_crop=auto_crop,
             crop_margin_mm=crop_margin_mm,
             metadata_extra=metadata_extra,
+            json_schema=json_schema,
             **kwargs,
         )
 
@@ -435,6 +441,9 @@ def save(
         return Path(spath)
         # return True
 
+    except AssertionError:
+        # Re-raise assertion errors - these are validation failures that should stop execution
+        raise
     except Exception as e:
         logger.error(
             f"Error occurred while saving: {str(e)}\n"
@@ -470,9 +479,7 @@ def _symlink_to(spath_final, symlink_to, verbose):
 
         # Ensure the symlink directory exists (only if there is a directory component)
         symlink_dir = _os.path.dirname(symlink_to)
-        if (
-            symlink_dir
-        ):  # Only create directory if there's a directory component
+        if symlink_dir:  # Only create directory if there's a directory component
             _os.makedirs(symlink_dir, exist_ok=True)
 
         # Remove existing symlink or file
@@ -483,9 +490,7 @@ def _symlink_to(spath_final, symlink_to, verbose):
 
         if verbose:
             symlink_to_full = (
-                os.path.realpath(symlink_to)
-                + "/"
-                + os.path.basename(spath_final)
+                os.path.realpath(symlink_to) + "/" + os.path.basename(spath_final)
             )
             logger.success(f"Symlinked: {spath_final} -> {symlink_to_full}")
 
@@ -501,6 +506,7 @@ def _save(
     auto_crop=False,
     crop_margin_mm=1.0,
     metadata_extra=None,
+    json_schema="recipe",
     **kwargs,
 ):
     # Don't use object's own save method - use consistent handlers
@@ -510,6 +516,11 @@ def _save(
     # Get file extension
     ext = _os.path.splitext(spath)[1].lower()
 
+    # Handle .canvas directories (special case - path ends with .canvas)
+    if spath.endswith(".canvas"):
+        save_canvas(obj, spath, **kwargs)
+        return
+
     # Try dispatch dictionary first for O(1) lookup
     if ext in _FILE_HANDLERS:
         # Check if handler needs special parameters
@@ -534,6 +545,7 @@ def _save(
                 auto_crop=auto_crop,
                 crop_margin_mm=crop_margin_mm,
                 metadata_extra=metadata_extra,
+                json_schema=json_schema,
                 **kwargs,
             )
         elif ext in [".hdf5", ".h5", ".zarr"]:
@@ -563,9 +575,7 @@ def _save(
             logger.success(f"Saved to: ./{rel_path} ({file_size})")
 
 
-def _save_separate_legends(
-    obj, spath, symlink_from_cwd=False, dry_run=False, **kwargs
-):
+def _save_separate_legends(obj, spath, symlink_from_cwd=False, dry_run=False, **kwargs):
     """Save separate legend files if ax.legend('separate') was used."""
     if dry_run:
         return
@@ -647,6 +657,7 @@ def _handle_image_with_csv(
     auto_crop=True,
     crop_margin_mm=1.0,
     metadata_extra=None,
+    json_schema="recipe",
     **kwargs,
 ):
     """Handle image file saving with optional CSV export and auto-cropping."""
@@ -655,7 +666,7 @@ def _handle_image_with_csv(
 
     # Auto-collect metadata from scitex figures if not explicitly provided
     collected_metadata = None
-    if 'metadata' not in kwargs or kwargs['metadata'] is None:
+    if "metadata" not in kwargs or kwargs["metadata"] is None:
         try:
             # Check if this is a matplotlib figure or scitex wrapper
             import matplotlib.figure
@@ -663,92 +674,154 @@ def _handle_image_with_csv(
             fig_mpl = None
             if isinstance(obj, matplotlib.figure.Figure):
                 fig_mpl = obj
-            elif hasattr(obj, '_fig_mpl'):  # FigWrapper
+            elif hasattr(obj, "_fig_mpl"):  # FigWrapper
                 fig_mpl = obj._fig_mpl
-            elif hasattr(obj, 'figure') and isinstance(obj.figure, matplotlib.figure.Figure):
+            elif hasattr(obj, "figure") and isinstance(
+                obj.figure, matplotlib.figure.Figure
+            ):
                 fig_mpl = obj.figure
 
             # If we have a figure, try to collect metadata
             if fig_mpl is not None:
-                # Get first axes if available
+                # Get axes from scitex wrapper if available (for multi-axes support)
+                # Priority: FigWrapper.axes (AxesWrapper) > mpl axes with _scitex_wrapper > mpl axes
                 ax = None
-                if hasattr(fig_mpl, 'axes') and len(fig_mpl.axes) > 0:
-                    ax = fig_mpl.axes[0]
+
+                # First try to get AxesWrapper from FigWrapper (obj)
+                if hasattr(obj, "axes"):
+                    # obj is FigWrapper, get its axes (could be AxisWrapper or AxesWrapper)
+                    ax = obj.axes
+                elif hasattr(fig_mpl, "axes") and len(fig_mpl.axes) > 0:
+                    mpl_ax = fig_mpl.axes[0]
+                    # Try to get scitex wrapper which has history for recipe schema
+                    if hasattr(mpl_ax, '_scitex_wrapper'):
+                        ax = mpl_ax._scitex_wrapper
+                    else:
+                        ax = mpl_ax
 
                 # Collect metadata using scitex's metadata collector
                 try:
-                    from scitex.plt.utils import collect_figure_metadata
-
-                    # Extract plot_id from filename (e.g., "01_plot.png" -> "01_plot")
-                    plot_id = _os.path.splitext(_os.path.basename(spath))[0]
-
-                    auto_metadata = collect_figure_metadata(fig_mpl, ax, plot_id=plot_id)
+                    if json_schema == "recipe":
+                        from scitex.plt.utils import collect_recipe_metadata
+                        auto_metadata = collect_recipe_metadata(
+                            fig_mpl, ax,
+                            auto_crop=auto_crop,
+                            crop_margin_mm=crop_margin_mm,
+                        )
+                    else:
+                        from scitex.plt.utils import collect_figure_metadata
+                        auto_metadata = collect_figure_metadata(fig_mpl, ax)
 
                     if auto_metadata:
-                        kwargs['metadata'] = auto_metadata
+                        kwargs["metadata"] = auto_metadata
                         collected_metadata = auto_metadata  # Save for JSON export
                         if verbose:
-                            logger.info("  • Auto-collected metadata from figure")
+                            schema_name = "recipe" if json_schema == "recipe" else "verbose"
+                            logger.info(f"  • Auto-collected metadata ({schema_name} schema)")
                 except ImportError:
                     pass  # collect_figure_metadata not available
                 except Exception as e:
                     if verbose:
                         import warnings
+
                         warnings.warn(f"Could not auto-collect metadata: {e}")
         except Exception:
             pass  # Silently continue if auto-collection fails
     else:
         # Use explicitly provided metadata
-        collected_metadata = kwargs.get('metadata')
+        collected_metadata = kwargs.get("metadata")
 
     # Merge metadata_extra with collected_metadata
     if metadata_extra is not None and collected_metadata is not None:
         # Deep merge: metadata_extra takes precedence
         import copy
+
         collected_metadata = copy.deepcopy(collected_metadata)
 
         # If metadata_extra has plot_type and it doesn't exist in collected, add it
-        if 'plot_type' in metadata_extra:
-            collected_metadata['plot_type'] = metadata_extra['plot_type']
+        if "plot_type" in metadata_extra:
+            collected_metadata["plot_type"] = metadata_extra["plot_type"]
 
         # Merge style information
-        if 'style' in metadata_extra:
-            collected_metadata['style'] = metadata_extra['style']
+        if "style" in metadata_extra:
+            collected_metadata["style"] = metadata_extra["style"]
 
         # Merge any other fields from metadata_extra
         for key, value in metadata_extra.items():
-            if key not in ['plot_type', 'style']:
+            if key not in ["plot_type", "style"]:
                 collected_metadata[key] = value
 
         # Update kwargs metadata for image saving
-        kwargs['metadata'] = collected_metadata
+        kwargs["metadata"] = collected_metadata
 
     save_image(obj, spath, verbose=verbose, **kwargs)
 
     # Auto-crop if requested (only for raster formats)
+    crop_offset = None
     if auto_crop and not dry_run:
         # Get file extension
         ext = spath.lower()
 
         # Only crop raster formats (PNG, JPEG, TIFF)
         # Skip vector formats (PDF, SVG) as they don't benefit from cropping
-        if ext.endswith(('.png', '.jpg', '.jpeg', '.tiff', '.tif')):
+        if ext.endswith((".png", ".jpg", ".jpeg", ".tiff", ".tif")):
             try:
                 from scitex.plt.utils._crop import crop
 
                 # Convert mm to pixels (assuming 300 DPI)
                 # 1mm at 300 DPI = 11.81 pixels ≈ 12 pixels
-                dpi = kwargs.get('dpi', 300)
+                dpi = kwargs.get("dpi", 300)
                 margin_px = int(crop_margin_mm * dpi / 25.4)  # 25.4mm per inch
 
-                # Crop the saved image in place
-                crop(spath, output_path=spath, margin=margin_px, overwrite=True, verbose=False)
+                # Crop the saved image in place, get crop offset for metadata adjustment
+                _, crop_offset = crop(
+                    spath,
+                    output_path=spath,
+                    margin=margin_px,
+                    overwrite=True,
+                    verbose=False,
+                    return_offset=True,
+                )
+
+                # Adjust axes_bbox_px in metadata to account for crop offset
+                if crop_offset and collected_metadata:
+                    if "axes_bbox_px" in collected_metadata:
+                        bbox = collected_metadata["axes_bbox_px"]
+                        # Subtract crop offset from all coordinates
+                        # left/upper is where the crop started
+                        left_offset = crop_offset["left"]
+                        upper_offset = crop_offset["upper"]
+                        bbox["x0"] = bbox.get("x0", 0) - left_offset
+                        bbox["x1"] = bbox.get("x1", 0) - left_offset
+                        bbox["y0"] = bbox.get("y0", 0) - upper_offset
+                        bbox["y1"] = bbox.get("y1", 0) - upper_offset
+                        # Update width/height to match new image size
+                        # (bbox width/height shouldn't change, but figure size does)
+
+                    # Also update figure size in metadata
+                    if "figure" in collected_metadata:
+                        fig_meta = collected_metadata["figure"]
+                        if "size_px" in fig_meta:
+                            fig_meta["size_px"] = [
+                                crop_offset["new_width"],
+                                crop_offset["new_height"],
+                            ]
+                    if "dimensions" in collected_metadata:
+                        dim_meta = collected_metadata["dimensions"]
+                        if "figure_size_px" in dim_meta:
+                            dim_meta["figure_size_px"] = [
+                                crop_offset["new_width"],
+                                crop_offset["new_height"],
+                            ]
 
                 if verbose:
-                    logger.info(f"  • Auto-cropped with {crop_margin_mm}mm margin ({margin_px}px at {dpi} DPI)")
+                    logger.info(
+                        f"  • Auto-cropped with {crop_margin_mm}mm margin ({margin_px}px at {dpi} DPI)"
+                    )
 
             except Exception as e:
                 import warnings
+
                 warnings.warn(f"Auto-crop failed: {e}. Image saved without cropping.")
 
     # Handle separate legend saving
@@ -766,11 +839,12 @@ def _handle_image_with_csv(
 
         # Check if the path contains an image extension directory (e.g., ./png/, ./jpg/)
         # If so, save CSV in a parallel ./csv/ directory
-        image_extensions = ['png', 'jpg', 'jpeg', 'gif', 'tiff', 'tif', 'svg', 'pdf']
+        image_extensions = ["png", "jpg", "jpeg", "gif", "tiff", "tif", "svg", "pdf"]
         parent_dir = _os.path.dirname(spath)
         parent_name = _os.path.basename(parent_dir)
         filename_without_ext = _os.path.splitext(_os.path.basename(spath))[0]
 
+        csv_path = None  # Initialize to avoid UnboundLocalError when CSV export is skipped
         try:
             # Get the figure object that may contain plot data
             fig_obj = _get_figure_with_data(obj)
@@ -786,7 +860,9 @@ def _handle_image_with_csv(
                             # Create parallel csv/ directory
                             grandparent_dir = _os.path.dirname(parent_dir)
                             csv_dir = _os.path.join(grandparent_dir, "csv")
-                            csv_path = _os.path.join(csv_dir, filename_without_ext + ".csv")
+                            csv_path = _os.path.join(
+                                csv_dir, filename_without_ext + ".csv"
+                            )
                         else:
                             # Save CSV in same directory as image
                             csv_path = _os.path.splitext(spath)[0] + ".csv"
@@ -804,20 +880,59 @@ def _handle_image_with_csv(
                             no_csv=True,
                         )
 
+                        # Update metadata with actual CSV info (after export)
+                        # This ensures column names match exactly, including any
+                        # deduplication suffixes added by pandas
+                        if collected_metadata is not None:
+                            try:
+                                from scitex.plt.utils._collect_figure_metadata import (
+                                    _compute_csv_hash,
+                                )
+
+                                # Ensure data section exists
+                                if "data" not in collected_metadata:
+                                    collected_metadata["data"] = {}
+
+                                # Get actual column names from exported DataFrame
+                                actual_columns = list(csv_data.columns)
+
+                                # Update data section with csv_path (relative to JSON)
+                                # Since JSON and CSV are in the same or parallel directories,
+                                # use just the filename for simplicity
+                                collected_metadata["data"]["csv_path"] = _os.path.basename(csv_path)
+
+                                # Update columns to use flat list of actual columns
+                                collected_metadata["data"]["columns_actual"] = actual_columns
+
+                                # Compute hash of actual CSV data
+                                collected_metadata["data"]["csv_hash"] = _compute_csv_hash(
+                                    csv_data
+                                )
+                            except Exception:
+                                pass  # Silently continue if update fails
+
                         # Create symlink_to for CSV if it was specified for the image
                         if symlink_to:
                             # Apply same directory transformation for symlink
                             symlink_parent_dir = _os.path.dirname(symlink_to)
                             symlink_parent_name = _os.path.basename(symlink_parent_dir)
-                            symlink_filename_without_ext = _os.path.splitext(_os.path.basename(symlink_to))[0]
+                            symlink_filename_without_ext = _os.path.splitext(
+                                _os.path.basename(symlink_to)
+                            )[0]
 
                             if symlink_parent_name.lower() in image_extensions:
-                                symlink_grandparent_dir = _os.path.dirname(symlink_parent_dir)
+                                symlink_grandparent_dir = _os.path.dirname(
+                                    symlink_parent_dir
+                                )
                                 csv_symlink_to = _os.path.join(
-                                    symlink_grandparent_dir, "csv", symlink_filename_without_ext + ".csv"
+                                    symlink_grandparent_dir,
+                                    "csv",
+                                    symlink_filename_without_ext + ".csv",
                                 )
                             else:
-                                csv_symlink_to = _os.path.splitext(symlink_to)[0] + ".csv"
+                                csv_symlink_to = (
+                                    _os.path.splitext(symlink_to)[0] + ".csv"
+                                )
 
                             _symlink_to(csv_path, csv_symlink_to, True)
 
@@ -836,14 +951,24 @@ def _handle_image_with_csv(
                                     ]
                                     if isinstance(original_path, str):
                                         # Apply same directory transformation for symlink
-                                        orig_parent_dir = _os.path.dirname(original_path)
-                                        orig_parent_name = _os.path.basename(orig_parent_dir)
-                                        orig_filename_without_ext = _os.path.splitext(_os.path.basename(original_path))[0]
+                                        orig_parent_dir = _os.path.dirname(
+                                            original_path
+                                        )
+                                        orig_parent_name = _os.path.basename(
+                                            orig_parent_dir
+                                        )
+                                        orig_filename_without_ext = _os.path.splitext(
+                                            _os.path.basename(original_path)
+                                        )[0]
 
                                         if orig_parent_name.lower() in image_extensions:
-                                            orig_grandparent_dir = _os.path.dirname(orig_parent_dir)
+                                            orig_grandparent_dir = _os.path.dirname(
+                                                orig_parent_dir
+                                            )
                                             csv_relative = _os.path.join(
-                                                orig_grandparent_dir, "csv", orig_filename_without_ext + ".csv"
+                                                orig_grandparent_dir,
+                                                "csv",
+                                                orig_filename_without_ext + ".csv",
                                             )
                                         else:
                                             csv_relative = original_path.replace(
@@ -859,9 +984,7 @@ def _handle_image_with_csv(
                             else:
                                 # Fallback to basename if we can't find the original path
                                 csv_cwd = (
-                                    _os.getcwd()
-                                    + "/"
-                                    + _os.path.basename(csv_path)
+                                    _os.getcwd() + "/" + _os.path.basename(csv_path)
                                 )
                                 _symlink(csv_path, csv_cwd, True, True)
 
@@ -873,14 +996,18 @@ def _handle_image_with_csv(
                         if parent_name.lower() in image_extensions:
                             grandparent_dir = _os.path.dirname(parent_dir)
                             csv_dir = _os.path.join(grandparent_dir, "csv")
-                            csv_sigmaplot_path = _os.path.join(csv_dir, filename_without_ext + "_for_sigmaplot.csv")
+                            csv_sigmaplot_path = _os.path.join(
+                                csv_dir, filename_without_ext + "_for_sigmaplot.csv"
+                            )
                         else:
                             csv_sigmaplot_path = spath.replace(
                                 ext_wo_dot, "csv"
                             ).replace(".csv", "_for_sigmaplot.csv")
 
                         # Ensure parent directory exists
-                        _os.makedirs(_os.path.dirname(csv_sigmaplot_path), exist_ok=True)
+                        _os.makedirs(
+                            _os.path.dirname(csv_sigmaplot_path), exist_ok=True
+                        )
                         # Save directly using _save to avoid path doubling
                         # Don't pass image-specific kwargs to CSV save
                         _save(
@@ -896,12 +1023,18 @@ def _handle_image_with_csv(
                         if symlink_to:
                             symlink_parent_dir = _os.path.dirname(symlink_to)
                             symlink_parent_name = _os.path.basename(symlink_parent_dir)
-                            symlink_filename_without_ext = _os.path.splitext(_os.path.basename(symlink_to))[0]
+                            symlink_filename_without_ext = _os.path.splitext(
+                                _os.path.basename(symlink_to)
+                            )[0]
 
                             if symlink_parent_name.lower() in image_extensions:
-                                symlink_grandparent_dir = _os.path.dirname(symlink_parent_dir)
+                                symlink_grandparent_dir = _os.path.dirname(
+                                    symlink_parent_dir
+                                )
                                 csv_sigmaplot_symlink_to = _os.path.join(
-                                    symlink_grandparent_dir, "csv", symlink_filename_without_ext + "_for_sigmaplot.csv"
+                                    symlink_grandparent_dir,
+                                    "csv",
+                                    symlink_filename_without_ext + "_for_sigmaplot.csv",
                                 )
                             else:
                                 csv_sigmaplot_symlink_to = (
@@ -938,7 +1071,16 @@ def _handle_image_with_csv(
             # If so, save JSON in a parallel ./json/ directory
             # Example: ./path/to/output/png/fig.png -> ./path/to/output/json/fig.json
             # Example: ./path/to/output/fig.png -> ./path/to/output/fig.json (same dir)
-            image_extensions = ['png', 'jpg', 'jpeg', 'gif', 'tiff', 'tif', 'svg', 'pdf']
+            image_extensions = [
+                "png",
+                "jpg",
+                "jpeg",
+                "gif",
+                "tiff",
+                "tif",
+                "svg",
+                "pdf",
+            ]
             parent_dir = _os.path.dirname(spath)
             parent_name = _os.path.basename(parent_dir)
             filename_without_ext = _os.path.splitext(_os.path.basename(spath))[0]
@@ -966,17 +1108,29 @@ def _handle_image_with_csv(
                 no_csv=True,
             )
 
+            # Verify CSV/JSON consistency (data_ref must match columns_actual)
+            # Only check for verbose schema - recipe schema uses different data_ref structure
+            if csv_path and not dry_run and json_schema != "recipe":
+                from scitex.plt.utils._collect_figure_metadata import (
+                    assert_csv_json_consistency,
+                )
+                assert_csv_json_consistency(csv_path, json_path)
+
             # Create symlink_to for JSON if it was specified for the image
             if symlink_to:
                 # Apply same directory transformation for symlink
                 symlink_parent_dir = _os.path.dirname(symlink_to)
                 symlink_parent_name = _os.path.basename(symlink_parent_dir)
-                symlink_filename_without_ext = _os.path.splitext(_os.path.basename(symlink_to))[0]
+                symlink_filename_without_ext = _os.path.splitext(
+                    _os.path.basename(symlink_to)
+                )[0]
 
                 if symlink_parent_name.lower() in image_extensions:
                     symlink_grandparent_dir = _os.path.dirname(symlink_parent_dir)
                     json_symlink_to = _os.path.join(
-                        symlink_grandparent_dir, "json", symlink_filename_without_ext + ".json"
+                        symlink_grandparent_dir,
+                        "json",
+                        symlink_filename_without_ext + ".json",
                     )
                 else:
                     json_symlink_to = _os.path.splitext(symlink_to)[0] + ".json"
@@ -998,12 +1152,16 @@ def _handle_image_with_csv(
                             # Apply same directory transformation for symlink
                             orig_parent_dir = _os.path.dirname(original_path)
                             orig_parent_name = _os.path.basename(orig_parent_dir)
-                            orig_filename_without_ext = _os.path.splitext(_os.path.basename(original_path))[0]
+                            orig_filename_without_ext = _os.path.splitext(
+                                _os.path.basename(original_path)
+                            )[0]
 
                             if orig_parent_name.lower() in image_extensions:
                                 orig_grandparent_dir = _os.path.dirname(orig_parent_dir)
                                 json_relative = _os.path.join(
-                                    orig_grandparent_dir, "json", orig_filename_without_ext + ".json"
+                                    orig_grandparent_dir,
+                                    "json",
+                                    orig_filename_without_ext + ".json",
                                 )
                             else:
                                 json_relative = original_path.replace(
@@ -1019,15 +1177,22 @@ def _handle_image_with_csv(
                     json_cwd = _os.getcwd() + "/" + _os.path.basename(json_path)
                     _symlink(json_path, json_cwd, True, True)
 
+        except AssertionError:
+            # Re-raise assertion errors - these are validation failures that should stop execution
+            raise
         except Exception as e:
             import warnings
+
             warnings.warn(f"JSON metadata export failed: {e}")
             import traceback
+
             traceback.print_exc()
 
 
 # Dispatch dictionary for O(1) file format lookup
 _FILE_HANDLERS = {
+    # Canvas directory format (scitex.vis)
+    ".canvas": save_canvas,
     # Excel formats
     ".xlsx": save_excel,
     ".xls": save_excel,

scitex/io/_save_modules/__init__.py

@@ -21,7 +21,10 @@ to that specific format.
 from ._csv import _save_csv as save_csv
 from ._excel import save_excel
 from ._numpy import _save_npy as save_npy, _save_npz as save_npz
-from ._pickle import _save_pickle as save_pickle, _save_pickle_gz as save_pickle_compressed
+from ._pickle import (
+    _save_pickle as save_pickle,
+    _save_pickle_gz as save_pickle_compressed,
+)
 from ._joblib import _save_joblib as save_joblib
 from ._torch import _save_torch as save_torch
 from ._json import _save_json as save_json
@@ -39,7 +42,9 @@ from ._bibtex import save_bibtex
 
 # Import additional save utilities
 from ._listed_dfs_as_csv import _save_listed_dfs_as_csv as save_listed_dfs_as_csv
-from ._listed_scalars_as_csv import _save_listed_scalars_as_csv as save_listed_scalars_as_csv
+from ._listed_scalars_as_csv import (
+    _save_listed_scalars_as_csv as save_listed_scalars_as_csv,
+)
 from ._optuna_study_as_csv_and_pngs import save_optuna_study_as_csv_and_pngs
 
 # Define what gets imported with "from scitex.io._save_modules import *"