MultiProcessRunner: Skips the tests that timeout on oss.

PiperOrigin-RevId: 337422385
Change-Id: Ia04297f82b6a51feb72fe5455d95942d5e170db2

tensorflow/python/distribute/multi_process_runner_test.py

@@ -319,6 +319,9 @@ class MultiProcessRunnerTest(test.TestCase):
 
   def test_seg_fault_raises_error(self):
 
+    if multi_process_runner.is_oss():
+      self.skipTest('TODO(b/171004637): Failing in OSS')
+
     def fn_expected_to_seg_fault():
       ctypes.string_at(0)  # Intentionally made seg fault.
 
@@ -336,6 +339,9 @@ class MultiProcessRunnerTest(test.TestCase):
 
   def test_seg_fault_in_chief_raises_error(self):
 
+    if multi_process_runner.is_oss():
+      self.skipTest('TODO(b/171004637): Failing in OSS')
+
     def fn_expected_to_seg_fault():
       if multi_worker_test_base.get_task_type() == 'worker':
         time.sleep(10000)
@@ -516,6 +522,9 @@ class MultiProcessRunnerTest(test.TestCase):
 
   def test_timeout_none(self):
 
+    if multi_process_runner.is_oss():
+      self.skipTest('Intentionally skipping longer test in OSS.')
+
     def fn():
       time.sleep(250)
       raise ValueError('Worker 0 errored')