wavedl 1.4.2__tar.gz → 1.4.3__tar.gz

{wavedl-1.4.2/src/wavedl.egg-info → wavedl-1.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: wavedl
-Version: 1.4.2
+Version: 1.4.3
 Summary: A Scalable Deep Learning Framework for Wave-Based Inverse Problems
 Author: Ductho Le
 License: MIT
@@ -57,6 +57,7 @@ Requires-Dist: triton>=2.0.0; sys_platform == "linux"
 [![Lint](https://img.shields.io/github/actions/workflow/status/ductho-le/WaveDL/lint.yml?branch=main&style=plastic&logo=ruff&logoColor=white&label=Lint)](https://github.com/ductho-le/WaveDL/actions/workflows/lint.yml)
 [![Try it on Colab](https://img.shields.io/badge/Try_it_on_Colab-8E44AD?style=plastic&logo=googlecolab&logoColor=white)](https://colab.research.google.com/github/ductho-le/WaveDL/blob/main/notebooks/demo.ipynb)
 <br>
+[![Downloads](https://img.shields.io/pepy/dt/wavedl?style=plastic&logo=pypi&logoColor=white&color=9ACD32)](https://pepy.tech/project/wavedl)
 [![License: MIT](https://img.shields.io/badge/License-MIT-orange.svg?style=plastic)](LICENSE)
 [![DOI](https://img.shields.io/badge/DOI-10.5281/zenodo.18012338-008080.svg?style=plastic)](https://doi.org/10.5281/zenodo.18012338)
 

{wavedl-1.4.2 → wavedl-1.4.3}/README.md

@@ -12,6 +12,7 @@
 [![Lint](https://img.shields.io/github/actions/workflow/status/ductho-le/WaveDL/lint.yml?branch=main&style=plastic&logo=ruff&logoColor=white&label=Lint)](https://github.com/ductho-le/WaveDL/actions/workflows/lint.yml)
 [![Try it on Colab](https://img.shields.io/badge/Try_it_on_Colab-8E44AD?style=plastic&logo=googlecolab&logoColor=white)](https://colab.research.google.com/github/ductho-le/WaveDL/blob/main/notebooks/demo.ipynb)
 <br>
+[![Downloads](https://img.shields.io/pepy/dt/wavedl?style=plastic&logo=pypi&logoColor=white&color=9ACD32)](https://pepy.tech/project/wavedl)
 [![License: MIT](https://img.shields.io/badge/License-MIT-orange.svg?style=plastic)](LICENSE)
 [![DOI](https://img.shields.io/badge/DOI-10.5281/zenodo.18012338-008080.svg?style=plastic)](https://doi.org/10.5281/zenodo.18012338)
 

{wavedl-1.4.2 → wavedl-1.4.3}/pyproject.toml

@@ -175,7 +175,7 @@ lines-after-imports = 2
 # Allow assert statements in tests
 "unit_tests/*" = ["B011", "S101"]
 # Allow unused imports for availability checks
-"src/wavedl/train.py" = ["F401"]
+"src/wavedl/train.py" = ["F401", "E402"]
 "src/wavedl/test.py" = ["F401"]
 "src/wavedl/utils/data.py" = ["F401"]
 

{wavedl-1.4.2 → wavedl-1.4.3}/src/wavedl/__init__.py

@@ -18,7 +18,7 @@ For inference:
     # or: python -m wavedl.test --checkpoint best_checkpoint --data_path test.npz
 """
 
-__version__ = "1.4.2"
+__version__ = "1.4.3"
 __author__ = "Ductho Le"
 __email__ = "ductho.le@outlook.com"
 

{wavedl-1.4.2 → wavedl-1.4.3}/src/wavedl/train.py

@@ -37,9 +37,54 @@ Author: Ductho Le (ductho.le@outlook.com)
 
 from __future__ import annotations
 
+# =============================================================================
+# HPC Environment Setup (MUST be before any library imports)
+# =============================================================================
+# Set writable cache directories for matplotlib and fontconfig ONLY when
+# the default paths are not writable (common on HPC clusters).
+import os
+import tempfile
+
+
+def _setup_cache_dir(env_var: str, default_subpath: str) -> None:
+    """Set cache directory only if default path is not writable."""
+    if env_var in os.environ:
+        return  # User already set, respect their choice
+
+    # Check if default home config path is writable
+    home = os.path.expanduser("~")
+    default_path = os.path.join(home, ".config", default_subpath)
+    default_parent = os.path.dirname(default_path)
+
+    # If default path or its parent is writable, let the library use defaults
+    if (
+        os.access(default_path, os.W_OK)
+        or (os.path.exists(default_parent) and os.access(default_parent, os.W_OK))
+        or os.access(os.path.join(home, ".config"), os.W_OK)
+    ):
+        return
+
+    # Default not writable - find alternative location
+    for cache_base in [
+        os.environ.get("SCRATCH"),
+        os.environ.get("SLURM_TMPDIR"),
+        tempfile.gettempdir(),
+    ]:
+        if cache_base and os.access(cache_base, os.W_OK):
+            cache_path = os.path.join(cache_base, f".{default_subpath}")
+            os.makedirs(cache_path, exist_ok=True)
+            os.environ[env_var] = cache_path
+            return
+
+
+_setup_cache_dir("MPLCONFIGDIR", "matplotlib")
+_setup_cache_dir("FONTCONFIG_CACHE", "fontconfig")
+
+# =============================================================================
+# Standard imports (after environment setup)
+# =============================================================================
 import argparse
 import logging
-import os
 import pickle
 import shutil
 import sys
@@ -47,6 +92,10 @@ import time
 import warnings
 from typing import Any
 
+
+# Suppress Pydantic warnings from accelerate's internal Field() usage
+warnings.filterwarnings("ignore", category=UserWarning, module="pydantic")
+
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
@@ -582,9 +631,9 @@ def main():
     # Torch 2.0 compilation (requires compatible Triton on GPU)
     if args.compile:
         try:
-            # Test if Triton is available AND compatible with this PyTorch version
-            # PyTorch needs triton_key from triton.compiler.compiler
-            from triton.compiler.compiler import triton_key
+            # Test if Triton is available - just import the package
+            # Different Triton versions have different internal APIs, so just check base import
+            import triton
 
             model = torch.compile(model)
             if accelerator.is_main_process:
@@ -875,9 +924,13 @@ def main():
             cpu_preds = torch.cat(local_preds)
             cpu_targets = torch.cat(local_targets)
 
-            # Gather to rank 0 only via gather_object (avoids all-gather to every rank)
-            # gather_object returns list of objects from each rank: [(preds0, targs0), (preds1, targs1), ...]
-            gathered = accelerator.gather_object((cpu_preds, cpu_targets))
+            # Gather predictions and targets across all ranks
+            # Use accelerator.gather (works with all accelerate versions)
+            gpu_preds = cpu_preds.to(accelerator.device)
+            gpu_targets = cpu_targets.to(accelerator.device)
+            all_preds_gathered = accelerator.gather(gpu_preds).cpu()
+            all_targets_gathered = accelerator.gather(gpu_targets).cpu()
+            gathered = [(all_preds_gathered, all_targets_gathered)]
 
             # Synchronize validation metrics (scalars only - efficient)
             val_loss_scalar = val_loss_sum.item()

{wavedl-1.4.2 → wavedl-1.4.3/src/wavedl.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: wavedl
-Version: 1.4.2
+Version: 1.4.3
 Summary: A Scalable Deep Learning Framework for Wave-Based Inverse Problems
 Author: Ductho Le
 License: MIT
@@ -57,6 +57,7 @@ Requires-Dist: triton>=2.0.0; sys_platform == "linux"
 [![Lint](https://img.shields.io/github/actions/workflow/status/ductho-le/WaveDL/lint.yml?branch=main&style=plastic&logo=ruff&logoColor=white&label=Lint)](https://github.com/ductho-le/WaveDL/actions/workflows/lint.yml)
 [![Try it on Colab](https://img.shields.io/badge/Try_it_on_Colab-8E44AD?style=plastic&logo=googlecolab&logoColor=white)](https://colab.research.google.com/github/ductho-le/WaveDL/blob/main/notebooks/demo.ipynb)
 <br>
+[![Downloads](https://img.shields.io/pepy/dt/wavedl?style=plastic&logo=pypi&logoColor=white&color=9ACD32)](https://pepy.tech/project/wavedl)
 [![License: MIT](https://img.shields.io/badge/License-MIT-orange.svg?style=plastic)](LICENSE)
 [![DOI](https://img.shields.io/badge/DOI-10.5281/zenodo.18012338-008080.svg?style=plastic)](https://doi.org/10.5281/zenodo.18012338)