[AIFlow] Fix test

flink-extended · Oct 15, 2021 · f6ed3d8 · f6ed3d8
1 parent d3581ab
commit f6ed3d8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ai_flow_plugins/blob_manager_plugins/oss_blob_manager.py b/ai_flow_plugins/blob_manager_plugins/oss_blob_manager.py
@@ -106,7 +106,10 @@ def download_project(self, workflow_snapshot_id, remote_path: Text, local_path:
                 logger.debug('Unlocked file {}'.format(lock_file_path))
                 lock_file.close()
                 if os.path.exists(lock_file_path):
-                    os.remove(lock_file_path)
+                    try:
+                        os.remove(lock_file_path)
+                    except OSError as e:
+                        logger.warning("Failed to remove lock file: {}".format(lock_file_path), exc_info=e)
         else:
             logger.info("Oss file: {} already exist at {}".format(oss_object_key, local_zip_file_path))