wavedl 1.4.2__py3-none-any.whl → 1.4.3__py3-none-any.whl

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/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.dist-info → wavedl-1.4.3.dist-info}/METADATA

@@ -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.dist-info → wavedl-1.4.3.dist-info}/RECORD

@@ -1,8 +1,8 @@
-wavedl/__init__.py,sha256=K52yq0nkj2B3W0ZpR3tb7RUcHMIiANAE2d1WTuGVZLI,1177
+wavedl/__init__.py,sha256=7uvgr9r21qLGu6RIZDUfQBhg1vXNAfmIWl_P6BMj_KQ,1177
 wavedl/hpc.py,sha256=de_GKERX8GS10sXRX9yXiGzMnk1jjq8JPzRw7QDs6d4,7967
 wavedl/hpo.py,sha256=YJXsnSGEBSVUqp_2ah7zu3_VClAUqZrdkuzDaSqQUjU,12952
 wavedl/test.py,sha256=jZmRJaivYYTMMTaccCi0yQjHOfp0a9YWR1wAPeKFH-k,36246
-wavedl/train.py,sha256=Gh02hlfjcote6w1sgUndJzKU_FhkJckTAhlq1aL8FY8,46842
+wavedl/train.py,sha256=TjLABBPCqu9r7FEWlxJlKsT7uAMo6hiDxRfii2SKXe4,49052
 wavedl/models/__init__.py,sha256=lfSohEnAUztO14nuwayMJhPjpgySzRN3jGiyAUuBmAU,3206
 wavedl/models/_template.py,sha256=J_D8taSPmV8lBaucN_vU-WiG98iFr7CJrZVNNX_Tdts,4600
 wavedl/models/base.py,sha256=T9iDF9IQM2MYucG_ggQd31rieUkB2fob-nkHyNIl2ak,7337
@@ -29,9 +29,9 @@ wavedl/utils/losses.py,sha256=5762M-TBC_hz6uyj1NPbU1vZeFOJQq7fR3-j7OygJRo,7254
 wavedl/utils/metrics.py,sha256=mkCpqZwl_XUpNvA5Ekjf7y-HqApafR7eR6EuA8cBdM8,37287
 wavedl/utils/optimizers.py,sha256=PyIkJ_hRhFi_Fio81Gy5YQNhcME0JUUEl8OTSyu-0RA,6323
 wavedl/utils/schedulers.py,sha256=e6Sf0yj8VOqkdwkUHLMyUfGfHKTX4NMr-zfgxWqCTYI,7659
-wavedl-1.4.2.dist-info/LICENSE,sha256=cEUCvcvH-9BT9Y-CNGY__PwWONCKu9zsoIqWA-NeHJ4,1066
-wavedl-1.4.2.dist-info/METADATA,sha256=xHMuRcGdF8Tdju6aBK601mqLYx0uSIdmtjoQrHpPYmI,40245
-wavedl-1.4.2.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
-wavedl-1.4.2.dist-info/entry_points.txt,sha256=f1RNDkXFZwBzrBzTMFocJ6xhfTvTmaEDTi5YyDEUaF8,140
-wavedl-1.4.2.dist-info/top_level.txt,sha256=ccneUt3D5Qzbh3bsBSSrq9bqrhGiogcWKY24ZC4Q6Xw,7
-wavedl-1.4.2.dist-info/RECORD,,
+wavedl-1.4.3.dist-info/LICENSE,sha256=cEUCvcvH-9BT9Y-CNGY__PwWONCKu9zsoIqWA-NeHJ4,1066
+wavedl-1.4.3.dist-info/METADATA,sha256=vs3nt8R2O5lD7q-si9M5ChyrriIKm0fFzDwi4HVIYxw,40386
+wavedl-1.4.3.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
+wavedl-1.4.3.dist-info/entry_points.txt,sha256=f1RNDkXFZwBzrBzTMFocJ6xhfTvTmaEDTi5YyDEUaF8,140
+wavedl-1.4.3.dist-info/top_level.txt,sha256=ccneUt3D5Qzbh3bsBSSrq9bqrhGiogcWKY24ZC4Q6Xw,7
+wavedl-1.4.3.dist-info/RECORD,,