remove print statement

Author: sophiayue1116

smdebug/tensorflow/keras.py

@@ -1105,7 +1105,7 @@ def unwrap(func):
 
     def close(self):
         self._cleanup()
-        print("\nStep Number in the close function: ", self.step)
+        # print("\nStep Number in the close function: ", self.step)
         if self.python_profiler:
             # print("python profiling for end of last train step to end of training")
             self.python_profiler.start_profiling(
@@ -1153,7 +1153,7 @@ def run(*args, **kwargs):
                 self.prepared_collections = True
 
             self._increment_step()
-            print("\nStep number in the push tape: ", self.step)
+            # print("\nStep number in the push tape: ", self.step)
 
             if self._get_collections_to_save_for_step():
                 # print('\n Collections saved for this step: ', self._get_collections_to_save_for_step())
@@ -1244,7 +1244,7 @@ def run(*args, **kwargs):
                 return
 
             self.last_saved_step = self.step
-            print("\nStep number in the pop tape: ", self.step)
+            # print("\nStep number in the pop tape: ", self.step)
 
         return run
 
@@ -1322,7 +1322,7 @@ def profiling_start_batch(self, mode):
             if self.mode != ModeKeys.GLOBAL:
                 self.mode_steps[ModeKeys.GLOBAL] = self.step
 
-        print("Step Number in start train batch: ", self.mode_steps[mode])
+        # print("Step Number in start train batch: ", self.mode_steps[mode])
 
         self.profiler_config_parser.load_config()
 
@@ -1385,7 +1385,7 @@ def profiling_end_batch(self, mode):
         """
         Enabling profiler at the end of train batch when native tf2 training is used.
         """
-        print("Step Number in end train batch: ", self.mode_steps[mode])
+        # print("Step Number in end train batch: ", self.mode_steps[mode])
 
         if self._is_not_supported():
             return
@@ -1421,7 +1421,7 @@ def profiling_end(self):
         """
         Stop profiler at the end of training when native tf2 training is used.
         """
-        print("Step Number at the end of training: ", self.step)
+        # print("Step Number at the end of training: ", self.step)
 
         # Unwrap the tape before closing and close the python profiling
         self.close()