GANDLF 0.1.5.dev20250813__py3-none-any.whl → 0.1.6.dev20250815__py3-none-any.whl

GANDLF/models/lightning_module.py

@@ -1713,7 +1713,7 @@ class GandlfLightningModule(pl.LightningModule):
             print(f"Previous latest model loaded from {self.model_paths['latest']}.")
         else:
             raise RuntimeError(
-                "No model found to load. Please train the model before running inference."
+                f"Best/latest models not found to load: {self.model_paths}"
             )
 
     @rank_zero_only

GANDLF/utils/modelio.py

@@ -175,7 +175,9 @@ def load_model(
     Returns:
         dict: Model dictionary containing model parameters and metadata.
     """
-    model_dict = torch.load(path, map_location=device)
+    # For PyTorch 2.7+, use weights_only=False to maintain backward compatibility
+    # This prevents the new security warning in PyTorch 2.7+
+    model_dict = torch.load(path, map_location=device, weights_only=False)
 
     # check if the model dictionary is complete
     if full_sanity_check:

GANDLF/version.py

@@ -1 +1 @@
-__version__ = "0.1.5-dev20250813"
+__version__ = "0.1.6-dev20250815"

{gandlf-0.1.5.dev20250813.dist-info → gandlf-0.1.6.dev20250815.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GANDLF
-Version: 0.1.5.dev20250813
+Version: 0.1.6.dev20250815
 Summary: PyTorch-based framework that handles segmentation/regression/classification using various DL architectures for medical imaging.
 Author: MLCommons
 Author-email: gandlf@mlcommons.org
@@ -18,9 +18,9 @@ Classifier: Topic :: Scientific/Engineering :: Medical Science Apps.
 Requires-Python: >3.9, <3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: torch==2.5.0
+Requires-Dist: torch==2.7.1
 Requires-Dist: black==23.11.0
-Requires-Dist: lightning==2.4.0
+Requires-Dist: lightning==2.5.2
 Requires-Dist: numpy==1.26.4
 Requires-Dist: scipy
 Requires-Dist: SimpleITK!=2.0.*

{gandlf-0.1.5.dev20250813.dist-info → gandlf-0.1.6.dev20250815.dist-info}/RECORD

@@ -5,7 +5,7 @@ GANDLF/logger.py,sha256=oamQ1SOTTpAnC8vQ67o211Q6_bExGg_hAuqlHGlJfAI,2951
 GANDLF/logging_config.yaml,sha256=9XxRxAKtLn5ehT1khpR8wEiJGW64sx1lylAspM5KaWk,1337
 GANDLF/parseConfig.py,sha256=jO-ybIPxLw23OWDvFdTukbft0ZM8UOofGnoL2C5CEps,754
 GANDLF/training_manager.py,sha256=AZlf-fl7KAwZgyre8-0M5lAyma6NvtiGX6XT51AJdxU,11436
-GANDLF/version.py,sha256=JOcwsj_97X0DG9YNnzo6sZ3JxhRbKQ6CtSkMFTMo1e0,34
+GANDLF/version.py,sha256=xm4CoShZnQ6gwuwJ2T05JSI9_mhJkSMinCm5nAGN4xQ,34
 GANDLF/anonymize/__init__.py,sha256=Nxig-jM-a-aKlK09PNi1zhNulEpLTyjnsY_oGQKdjhQ,1953
 GANDLF/anonymize/convert_to_nifti.py,sha256=MOfSDncFGJGb-EQP9sFGn0yuKpX010Ioi2KNwttaex8,1339
 GANDLF/cli/__init__.py,sha256=F05eyL2HKyWkHczRZuPE_Z2Yg685P9ARYxTwz6njGeQ,784
@@ -125,7 +125,7 @@ GANDLF/models/imagenet_unet.py,sha256=vTAuckJ0Bs9nfe2G0nKISqgxWArIYdM4poGpTk56QY
 GANDLF/models/imagenet_vgg.py,sha256=FrMNLIxCtaMHWZ9JO7pWtoXTT1f8eVDuREx9108Aw2g,10367
 GANDLF/models/light_unet.py,sha256=jzcTHH7kv_ikriYxIe7fbQJkmwOQZzePW4KLCW2FGWM,6645
 GANDLF/models/light_unet_multilayer.py,sha256=xrKy--lljbCX7U2eYjpFZA69C6qlHmA5W-fEckPBll4,4534
-GANDLF/models/lightning_module.py,sha256=kZhlCSD_JLJFFqVRGUEz1CgvENWHqVgc418_fvbQdkk,83347
+GANDLF/models/lightning_module.py,sha256=QLBL-QKp5DADHn8PpGjAJO4l3J7M4WI95kymPFvfGOk,83332
 GANDLF/models/modelBase.py,sha256=NFmKcEylNchm0LVqwRoD_iIMFj36IsaKwVCPJPLux3g,5752
 GANDLF/models/resnet.py,sha256=0PB7sGSLHUq2F6tyHYqxepmU1eBccSwCzBrD7Qmc280,16987
 GANDLF/models/sdnet.py,sha256=uhHvOHRTAx4ANxv2lKXfBn6Bg4Broqzb77qekVjXeFI,14528
@@ -177,14 +177,14 @@ GANDLF/utils/generic.py,sha256=NyJgaeCdryBlHMziPPLmWtIk450u042NWQgs_qh7knw,11744
 GANDLF/utils/handle_collisions.py,sha256=UFgfrddQqeipOdpKk0Mo-wwJ-TYTESNZyXzEyQeaLdI,3082
 GANDLF/utils/imaging.py,sha256=_qCgvIXHZW_k_IUYLU95G5L5yoozxCRCYxo65TI-95U,15983
 GANDLF/utils/modelbase.py,sha256=MJ1ufJGQ-ZgfYWTg7o4f-Iz78d7SYiyxN7MT8bnNDaw,950
-GANDLF/utils/modelio.py,sha256=9koLUn3wCODtNOYA2abtxkqxfLgEG07Bhh-NYIrtWOc,8084
+GANDLF/utils/modelio.py,sha256=xmzAn7P0kqJd_Gcf17nMcV784poW2qpi3AY483Tk2ko,8247
 GANDLF/utils/parameter_processing.py,sha256=DA7ZEsizWWLJZnCxBnDNh1NyA-bw5oirOvodiZZWbIk,5675
 GANDLF/utils/pred_target_processors.py,sha256=aatXJ6jdJaNAHa_tPzHfC1gOQrPYJLtg-cYeUFvkM_s,2701
 GANDLF/utils/tensor.py,sha256=SbcTyWs_qaFTnxpFlJHiq9xNKesDu7t0YZsm9eUnDSk,21565
 GANDLF/utils/write_parse.py,sha256=HROxskhet-uIdUJmz16z_7p9r0mf8hWsqQFygwZ8ap0,9202
-gandlf-0.1.5.dev20250813.dist-info/licenses/LICENSE,sha256=GlZPAfA4eckod8IVayhBXkqCpESXf6cc9BGli_Jwims,11357
-gandlf-0.1.5.dev20250813.dist-info/METADATA,sha256=PZRRxKt5TEz_h8MZNUcvUDpBykl4LIneYWz4E0p3J9E,9904
-gandlf-0.1.5.dev20250813.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gandlf-0.1.5.dev20250813.dist-info/entry_points.txt,sha256=agwocNI7Upi-sFDe1rMl71dGN8YhCBB7WJmtBHRF4jg,902
-gandlf-0.1.5.dev20250813.dist-info/top_level.txt,sha256=i5D9wEbQhl085_9Lx2m7x-9Zu6nlx1tjYYbuSihG09E,7
-gandlf-0.1.5.dev20250813.dist-info/RECORD,,
+gandlf-0.1.6.dev20250815.dist-info/licenses/LICENSE,sha256=GlZPAfA4eckod8IVayhBXkqCpESXf6cc9BGli_Jwims,11357
+gandlf-0.1.6.dev20250815.dist-info/METADATA,sha256=rVGV3dzH_WwV-a_YY2n5h-Wi1X0WBBkYbHSTpJcBJKY,9904
+gandlf-0.1.6.dev20250815.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gandlf-0.1.6.dev20250815.dist-info/entry_points.txt,sha256=agwocNI7Upi-sFDe1rMl71dGN8YhCBB7WJmtBHRF4jg,902
+gandlf-0.1.6.dev20250815.dist-info/top_level.txt,sha256=i5D9wEbQhl085_9Lx2m7x-9Zu6nlx1tjYYbuSihG09E,7
+gandlf-0.1.6.dev20250815.dist-info/RECORD,,