Update logger.py

samikama · web-flow · commit 381d88f67ce1 · 2018-09-12T19:48:32.000-07:00
Spaces after commas
diff --git a/official/utils/logs/logger.py b/official/utils/logs/logger.py
@@ -315,7 +315,7 @@ def _gather_run_info(model_name, dataset_name, run_params, test_id):
   _collect_tensorflow_environment_variables(run_info)
   _collect_run_params(run_info, run_params)
   _collect_cpu_info(run_info)
-  _collect_gpu_info(run_info,session_config)
+  _collect_gpu_info(run_info, session_config)
   _collect_memory_info(run_info)
   _collect_test_environment(run_info)
   return run_info
@@ -389,7 +389,7 @@ def _collect_cpu_info(run_info):
     tf.logging.warn("'cpuinfo' not imported. CPU info will not be logged.")
 
 
-def _collect_gpu_info(run_info,session_config=None):
+def _collect_gpu_info(run_info, session_config=None):
   """Collect local GPU information by TF device library."""
   gpu_info = {}
   local_device_protos = device_lib.list_local_devices(session_config)
