unable to build Docker container for SageMaker

[tohara-PandoLogic]

We are unable to build Docker container for SageMaker following the steps under
https://aws.amazon.com/blogs/machine-learning/train-albert-for-natural-language-processing-with-tensorflow-on-amazon-sagemaker.

Specifically, it fails to checkout jarednielsen/transformers.git@tfsquad:

$ pip install git+git://github.com/jarednielsen/transformers.git@tfsquad
Collecting git+git://github.com/jarednielsen/transformers.git@tfsquad
  Cloning git://github.com/jarednielsen/transformers.git (to revision tfsquad) to ./temp/tmp/pip-req-build-2ug44ime
  Running command git clone --filter=blob:none --quiet git://github.com/jarednielsen/transformers.git /Users/tohara/temp/tmp/pip-req-build-2ug44ime
  fatal: unable to connect to github.com:
  github.com[0: 140.82.112.4]: errno=Operation timed out
  error: subprocess-exited-with-error

from in ./models/nlp/docker/ngc_sagemaker.Dockerfile:

:RUN pip install git+git://github.com/jarednielsen/transformers.git@tfsquad

The Github connection is fine as the following works:

pip install git+https://github.com/benjaminp/six

A colleague ran into this problem, and I reproduced the problem in a separate environment.

Best,
Tom