weights_only=False for hydra

matsumotosan · matsumotosan · commit 7ddb4f80fc21 · 2025-08-21T00:49:35.000-04:00
diff --git a/tests/tests_pytorch/models/test_hparams.py b/tests/tests_pytorch/models/test_hparams.py
@@ -791,7 +791,7 @@ def __init__(self, args_0, args_1, args_2, kwarg_1=None):
             logger=False,
         )
         trainer.fit(model)
-        _ = TestHydraModel.load_from_checkpoint(checkpoint_callback.best_model_path)
+        _ = TestHydraModel.load_from_checkpoint(checkpoint_callback.best_model_path, weights_only=False)
 
 
 @pytest.mark.parametrize("ignore", ["arg2", ("arg2", "arg3")])

Original file line number	Diff line number	Diff line change
`@@ -791,7 +791,7 @@ def __init__(self, args_0, args_1, args_2, kwarg_1=None):`
`791`	`791`	`logger=False,`
`792`	`792`	`)`
`793`	`793`	`trainer.fit(model)`
`794`		`- _ = TestHydraModel.load_from_checkpoint(checkpoint_callback.best_model_path)`
	`794`	`+ _ = TestHydraModel.load_from_checkpoint(checkpoint_callback.best_model_path, weights_only=False)`
`795`	`795`
`796`	`796`
`797`	`797`	`@pytest.mark.parametrize("ignore", ["arg2", ("arg2", "arg3")])`