From ba24e030fe0b8231b1940243cb1b5c273c169713 Mon Sep 17 00:00:00 2001 From: Sean Yang Date: Thu, 10 Oct 2024 09:35:32 -0700 Subject: [PATCH] add none default values --- nvflare/app_opt/pt/job_config/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nvflare/app_opt/pt/job_config/model.py b/nvflare/app_opt/pt/job_config/model.py index 5d6782089e..d050148cf8 100644 --- a/nvflare/app_opt/pt/job_config/model.py +++ b/nvflare/app_opt/pt/job_config/model.py @@ -24,7 +24,7 @@ class PTModel: - def __init__(self, model, persistor: Optional[ModelPersistor], locator: Optional[ModelLocator]): + def __init__(self, model, persistor: Optional[ModelPersistor] = None, locator: Optional[ModelLocator] = None): """PyTorch model wrapper. If model is an nn.Module, add a PTFileModelPersistor with the model and a TFModelPersistor.