def mock_pg()

in common/testing_utils.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

def mock_pg():
  with patch.dict(os.environ, MOCK_ENV):
    try:
      dist.init_process_group(
        backend="gloo",
        timeout=datetime.timedelta(1),
      )
      yield
    except:
      dist.destroy_process_group()
      raise
    finally:
      dist.destroy_process_group()