Skip to content

Commit 2b028a3

Browse files
authored
[Estimator] Fix Estimator export_savedmodel to oss. (#6)
1 parent b0c0123 commit 2b028a3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensorflow_estimator/python/estimator/estimator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -880,7 +880,8 @@ def _export_all_saved_models(
880880
gfile.MakeDirs(dest_path)
881881
gfile.Copy(source, dest_absolute)
882882

883-
gfile.Rename(temp_export_dir, export_dir)
883+
# '/' is needed when rename a oss directory.
884+
gfile.Rename(temp_export_dir + '/', export_dir)
884885
return export_dir
885886

886887
def _add_meta_graph_for_mode(self,

0 commit comments

Comments
 (0)