add weights_save_path to deprecated args

Signed-off-by: nithinraok <[email protected]>
NVIDIA · Sep 12, 2022 · 93e04b3 · 93e04b3
1 parent 3200c95
commit 93e04b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/core/test_config_utils.py b/tests/core/test_config_utils.py
@@ -115,7 +115,7 @@ class DummyDataClass:
 
     @pytest.mark.unit
     def test_ptl_config(self):
-        PTL_DEPRECATED = ['distributed_backend', 'automatic_optimization', 'gpus', 'num_processes']
+        PTL_DEPRECATED = ['distributed_backend', 'automatic_optimization', 'gpus', 'num_processes', 'weights_save_path']
 
         result = config_utils.assert_dataclass_signature_match(ptl.Trainer, TrainerConfig, ignore_args=PTL_DEPRECATED)
         signatures_match, cls_subset, dataclass_subset = result