message: Enable multi_worker_callback_tf2_test.py in OSS.
diff: @@ -20,7 +20,6 @@ from __future__ import print_function
  import json
  import os
  -import sys
  from absl.testing import parameterized
  @@ -36,11 +35,6 @@ from tensorflow.python.lib.io import file_io
  from tensorflow.python.platform import test
  -def _is_oss():
  - """Returns whether the test is run under OSS."""
  - return len(sys.argv) >= 1 and 'bazel' in sys.argv[0]
  -
  -
  def checkpoint_exists(filepath):
  """Returns whether the checkpoint `filepath` refers to exists."""
  if filepath.endswith('.h5'):
  @@ -189,8 +183,6 @@ class KerasCallbackMultiProcessTest(parameterized.TestCase, test.TestCase):
  def proc_model_checkpoint_works_with_same_file_path(test_obj,
  saving_filepath):
  - if _is_oss():
  - test_obj.skipTest('TODO(b/170838633): Failing in OSS')
  model, _, train_ds, steps = _model_setup(test_obj, file_format='')
  num_epoch = 4
  