Fix session_debug_test.py (patched from #754)

Author: nfelt

tensorboard/plugins/debugger/session_debug_test.py

@@ -205,15 +205,13 @@ def testRunSimpleNetworkoWithInfAndNaNWorks(self):
 
     report = self._debug_data_server.numerics_alert_report()
     self.assertEqual(2, len(report))
-    self.assertEqual("/job:localhost/replica:0/task:0/cpu:0",
-                     report[0].device_name)
+    self.assertTrue(report[0].device_name.lower().endswith("cpu:0"))
     self.assertEqual("u:0", report[0].tensor_name)
     self.assertGreater(report[0].first_timestamp, 0)
     self.assertEqual(0, report[0].nan_event_count)
     self.assertEqual(0, report[0].neg_inf_event_count)
     self.assertEqual(1, report[0].pos_inf_event_count)
-    self.assertEqual("/job:localhost/replica:0/task:0/cpu:0",
-                     report[1].device_name)
+    self.assertTrue(report[1].device_name.lower().endswith("cpu:0"))
     self.assertEqual("u:0", report[0].tensor_name)
     self.assertGreaterEqual(report[1].first_timestamp,
                             report[0].first_timestamp)
@@ -299,7 +297,7 @@ def testConcurrentNumericsAlertsAreRegisteredCorrectly(self):
 
       def run_v(thread_id):
         for _ in range(num_runs_per_thread):
-          sess.run(v, options=run_options_list[thread_id])  # DEBUG
+          sess.run(v, options=run_options_list[thread_id])
 
       run_threads = []
       for thread_id in range(num_threads):
@@ -312,15 +310,13 @@ def run_v(thread_id):
 
     report = self._debug_data_server.numerics_alert_report()
     self.assertEqual(2, len(report))
-    self.assertEqual("/job:localhost/replica:0/task:0/cpu:0",
-                     report[0].device_name)
+    self.assertTrue(report[0].device_name.lower().endswith("cpu:0"))
     self.assertEqual("u:0", report[0].tensor_name)
     self.assertGreater(report[0].first_timestamp, 0)
     self.assertEqual(0, report[0].nan_event_count)
     self.assertEqual(0, report[0].neg_inf_event_count)
     self.assertEqual(total_num_runs, report[0].pos_inf_event_count)
-    self.assertEqual("/job:localhost/replica:0/task:0/cpu:0",
-                     report[1].device_name)
+    self.assertTrue(report[1].device_name.lower().endswith("cpu:0"))
     self.assertEqual("u:0", report[0].tensor_name)
     self.assertGreaterEqual(report[1].first_timestamp,
                             report[0].first_timestamp)