Add Returns section to get_generated_files def

tensorflow · Nov 18, 2020 · 10335ec · 10335ec
1 parent bc2e521
commit 10335ec
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/python/tensorflow_cloud/core/containerize.py b/src/python/tensorflow_cloud/core/containerize.py
@@ -124,6 +124,10 @@ def get_generated_files(self, return_descriptors=False):
 
         Args:
             return_descriptors: Whether to return descriptors as well.
+        
+        Returns:
+          Docker and tar file paths. Depending on return_descriptors, possibly
+          their file descriptors as well. 
         """
         if return_descriptors:
           return [