Load with weights_only=True

dftd3 · Sep 19, 2024 · ac2e91d · ac2e91d
1 parent d928ba4
commit ac2e91d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/tad_dftd3/data/radii.py b/src/tad_dftd3/data/radii.py
@@ -52,7 +52,7 @@ def _load_vdw_rad_d3(
         VDW radii.
     """
     path = op.join(op.dirname(__file__), "vdw-d3.pt")
-    return torch.load(path).type(dtype).to(device)
+    return torch.load(path, weights_only=True).to(device=device, dtype=dtype)
 
 
 VDW_D3 = _load_vdw_rad_d3()
diff --git a/src/tad_dftd3/reference.py b/src/tad_dftd3/reference.py
@@ -178,7 +178,7 @@ def _load_c6(
         Reference C6 coefficients.
     """
     path = op.join(op.dirname(__file__), "reference-c6.pt")
-    return torch.load(path).type(dtype).to(device)
+    return torch.load(path, weights_only=True).to(device=device, dtype=dtype)
 
 
 class Reference: