wavedl 1.6.2__tar.gz → 1.6.3__tar.gz

{wavedl-1.6.2/src/wavedl.egg-info → wavedl-1.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: wavedl
-Version: 1.6.2
+Version: 1.6.3
 Summary: A Scalable Deep Learning Framework for Wave-Based Inverse Problems
 Author: Ductho Le
 License: MIT
@@ -214,11 +214,11 @@ This installs everything you need: training, inference, HPO, ONNX export.
 ```bash
 git clone https://github.com/ductho-le/WaveDL.git
 cd WaveDL
-pip install -e .
+pip install -e ".[dev]"
 ```
 
 > [!NOTE]
-> Python 3.11+ required. For development setup, see [CONTRIBUTING.md](.github/CONTRIBUTING.md).
+> Python 3.11+ required. For contributor setup (pre-commit hooks), see [CONTRIBUTING.md](.github/CONTRIBUTING.md).
 
 ### Quick Start
 
@@ -273,8 +273,6 @@ accelerate launch --num_machines 2 --main_process_ip <ip> -m wavedl.train --mode
 
 ### Testing & Inference
 
-After training, use `wavedl-test` to evaluate your model on test data:
-
 ```bash
 # Basic inference
 wavedl-test --checkpoint <checkpoint_folder> --data_path <test_data>

{wavedl-1.6.2 → wavedl-1.6.3}/README.md

@@ -166,11 +166,11 @@ This installs everything you need: training, inference, HPO, ONNX export.
 ```bash
 git clone https://github.com/ductho-le/WaveDL.git
 cd WaveDL
-pip install -e .
+pip install -e ".[dev]"
 ```
 
 > [!NOTE]
-> Python 3.11+ required. For development setup, see [CONTRIBUTING.md](.github/CONTRIBUTING.md).
+> Python 3.11+ required. For contributor setup (pre-commit hooks), see [CONTRIBUTING.md](.github/CONTRIBUTING.md).
 
 ### Quick Start
 
@@ -225,8 +225,6 @@ accelerate launch --num_machines 2 --main_process_ip <ip> -m wavedl.train --mode
 
 ### Testing & Inference
 
-After training, use `wavedl-test` to evaluate your model on test data:
-
 ```bash
 # Basic inference
 wavedl-test --checkpoint <checkpoint_folder> --data_path <test_data>

{wavedl-1.6.2 → wavedl-1.6.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.6.2"
+__version__ = "1.6.3"
 __author__ = "Ductho Le"
 __email__ = "ductho.le@outlook.com"
 

{wavedl-1.6.2 → wavedl-1.6.3}/src/wavedl/models/__init__.py

@@ -77,6 +77,15 @@ from .unet import UNetRegression
 from .vit import ViTBase_, ViTSmall, ViTTiny
 
 
+# Optional RATENet (unpublished, may be gitignored)
+try:
+    from .ratenet import RATENet, RATENetLite, RATENetTiny, RATENetV2
+
+    _HAS_RATENET = True
+except ImportError:
+    _HAS_RATENET = False
+
+
 # Optional timm-based models (imported conditionally)
 try:
     from .caformer import CaFormerS18, CaFormerS36, PoolFormerS12
@@ -111,6 +120,7 @@ __all__ = [
     "MC3_18",
     "MODEL_REGISTRY",
     "TCN",
+    # Classes (uppercase first, alphabetically)
     "BaseModel",
     "ConvNeXtBase_",
     "ConvNeXtSmall",
@@ -152,6 +162,7 @@ __all__ = [
     "VimBase",
     "VimSmall",
     "VimTiny",
+    # Functions (lowercase, alphabetically)
     "build_model",
     "get_model",
     "list_models",
@@ -186,3 +197,14 @@ if _HAS_TIMM_MODELS:
             "UniRepLKNetTiny",
         ]
     )
+
+# Add RATENet models to __all__ if available (unpublished)
+if _HAS_RATENET:
+    __all__.extend(
+        [
+            "RATENet",
+            "RATENetLite",
+            "RATENetTiny",
+            "RATENetV2",
+        ]
+    )

{wavedl-1.6.2 → wavedl-1.6.3}/src/wavedl/test.py

@@ -398,6 +398,14 @@ def load_checkpoint(
 
     if HAS_SAFETENSORS and weight_path.suffix == ".safetensors":
         state_dict = load_safetensors(str(weight_path))
+    elif weight_path.suffix == ".safetensors":
+        # Safetensors file exists but library not installed
+        raise ImportError(
+            f"Checkpoint uses safetensors format ({weight_path.name}) but "
+            f"'safetensors' package is not installed. Install it with:\n"
+            f"    pip install safetensors\n"
+            f"Or convert the checkpoint to PyTorch format (model.bin)."
+        )
     else:
         state_dict = torch.load(weight_path, map_location="cpu", weights_only=True)
 

{wavedl-1.6.2 → wavedl-1.6.3}/src/wavedl/train.py

@@ -1266,10 +1266,10 @@ def main():
                         os.path.join(args.output_dir, "best_model_weights.pth"),
                     )
 
-                    # Copy scaler to checkpoint for portability
+                    # Copy scaler to checkpoint for portability (always overwrite to stay current)
                     scaler_src = os.path.join(args.output_dir, "scaler.pkl")
                     scaler_dst = os.path.join(ckpt_dir, "scaler.pkl")
-                    if os.path.exists(scaler_src) and not os.path.exists(scaler_dst):
+                    if os.path.exists(scaler_src):
                         shutil.copy2(scaler_src, scaler_dst)
 
                     logger.info(

{wavedl-1.6.2 → wavedl-1.6.3}/src/wavedl/utils/data.py

@@ -984,7 +984,15 @@ def load_test_data(
                 f"Available keys depend on file format. Original error: {e}"
             ) from e
 
-        # Legitimate fallback: no explicit output_key, outputs just not present
+        # Also fail-fast if explicit input_key was provided but not found
+        # This prevents silently loading a different tensor when user mistyped key
+        if input_key is not None:
+            raise KeyError(
+                f"Explicit --input_key '{input_key}' not found in file. "
+                f"Original error: {e}"
+            ) from e
+
+        # Legitimate fallback: no explicit keys, outputs just not present
         if format == "npz":
             # First pass to find keys
             with np.load(path, allow_pickle=False) as probe:
@@ -1524,21 +1532,43 @@ def prepare_data(
 
             logger.info("   ✔ Cache creation complete, synchronizing ranks...")
         else:
-            # NON-MAIN RANKS: Wait for cache creation
-            # Log that we're waiting (helps with debugging)
+            # NON-MAIN RANKS: Wait for cache creation with timeout
+            # Use monotonic clock (immune to system clock changes)
             import time
 
-            wait_start = time.time()
+            wait_start = time.monotonic()
+
+            # Robust env parsing with guards for invalid/non-positive values
+            DEFAULT_CACHE_TIMEOUT = 3600  # 1 hour default
+            try:
+                env_timeout = os.environ.get("WAVEDL_CACHE_TIMEOUT", "")
+                CACHE_TIMEOUT = (
+                    int(env_timeout) if env_timeout else DEFAULT_CACHE_TIMEOUT
+                )
+                if CACHE_TIMEOUT <= 0:
+                    CACHE_TIMEOUT = DEFAULT_CACHE_TIMEOUT
+            except ValueError:
+                CACHE_TIMEOUT = DEFAULT_CACHE_TIMEOUT
+
             while not (
                 os.path.exists(CACHE_FILE)
                 and os.path.exists(SCALER_FILE)
                 and os.path.exists(META_FILE)
             ):
                 time.sleep(5)  # Check every 5 seconds
-                elapsed = time.time() - wait_start
+                elapsed = time.monotonic() - wait_start
+
+                if elapsed > CACHE_TIMEOUT:
+                    raise RuntimeError(
+                        f"[Rank {accelerator.process_index}] Timeout waiting for cache "
+                        f"files after {CACHE_TIMEOUT}s. Rank 0 may have failed during "
+                        f"cache generation. Check rank 0 logs for errors."
+                    )
+
                 if elapsed > 60 and int(elapsed) % 60 < 5:  # Log every ~minute
                     logger.info(
-                        f"   [Rank {accelerator.process_index}] Waiting for cache creation... ({int(elapsed)}s)"
+                        f"   [Rank {accelerator.process_index}] Waiting for cache "
+                        f"creation... ({int(elapsed)}s / {CACHE_TIMEOUT}s max)"
                     )
             # Small delay to ensure files are fully written
             time.sleep(2)

{wavedl-1.6.2 → wavedl-1.6.3/src/wavedl.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: wavedl
-Version: 1.6.2
+Version: 1.6.3
 Summary: A Scalable Deep Learning Framework for Wave-Based Inverse Problems
 Author: Ductho Le
 License: MIT
@@ -214,11 +214,11 @@ This installs everything you need: training, inference, HPO, ONNX export.
 ```bash
 git clone https://github.com/ductho-le/WaveDL.git
 cd WaveDL
-pip install -e .
+pip install -e ".[dev]"
 ```
 
 > [!NOTE]
-> Python 3.11+ required. For development setup, see [CONTRIBUTING.md](.github/CONTRIBUTING.md).
+> Python 3.11+ required. For contributor setup (pre-commit hooks), see [CONTRIBUTING.md](.github/CONTRIBUTING.md).
 
 ### Quick Start
 
@@ -273,8 +273,6 @@ accelerate launch --num_machines 2 --main_process_ip <ip> -m wavedl.train --mode
 
 ### Testing & Inference
 
-After training, use `wavedl-test` to evaluate your model on test data:
-
 ```bash
 # Basic inference
 wavedl-test --checkpoint <checkpoint_folder> --data_path <test_data>