[Estimator] Fix Estimator export_savedmodel to oss. (#7)

DeepRec-AI · Nov 15, 2022 · 7cb9f95 · 7cb9f95
1 parent b0c0123
commit 7cb9f95
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow_estimator/python/estimator/estimator.py b/tensorflow_estimator/python/estimator/estimator.py
@@ -880,7 +880,8 @@ def _export_all_saved_models(
           gfile.MakeDirs(dest_path)
           gfile.Copy(source, dest_absolute)
 
-      gfile.Rename(temp_export_dir, export_dir)
+      # '/' is needed when rename a oss directory.
+      gfile.Rename(temp_export_dir + b'/', export_dir)
       return export_dir
 
   def _add_meta_graph_for_mode(self,