add none default values (#3025)

NVIDIA · Oct 10, 2024 · 57d843a · 57d843a
1 parent 3695979
commit 57d843a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.