From 57d843ab77756fb524172500fc1fff0a5999c242 Mon Sep 17 00:00:00 2001
From: Sean Yang <seany314@gmail.com>
Date: Thu, 10 Oct 2024 11:01:04 -0700
Subject: [PATCH] add none default values (#3025)

---
 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.