handle artifacts when path is dir (#6658)

Signed-off-by: arendu <[email protected]>
NVIDIA · May 16, 2023 · cfff834 · cfff834
1 parent 90156b1
commit cfff834
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/nemo/core/connectors/save_restore_connector.py b/nemo/core/connectors/save_restore_connector.py
@@ -470,7 +470,10 @@ def _handle_artifacts(self, model, nemo_file_folder):
                     # unpack all restorations paths (nemo checkpoints)
                     # in nemo checkpoints all resources contain hash in name, so there should be no collisions
                     for path in restoration_paths:
-                        self._unpack_nemo_file(path2file=path, out_folder=archive_dir)
+                        if self.model_extracted_dir:
+                            shutil.copytree(src=path, dst=archive_dir, dirs_exist_ok=True)
+                        else:
+                            self._unpack_nemo_file(path2file=path, out_folder=archive_dir)
                     os.chdir(archive_dir)
                     for conf_path, artiitem in tarfile_artifacts:
                         # Get basename and copy it to nemo_file_folder