Sagemaker huggingface estimator tries to import tensorflow when pytorch is defined

I have problems running training jobs on sagemaker using the Huggingface estimator because it tries to import tensorflow even though pytorch is the defined framework. Did anyone face the same issue? Issue on Github contains all the details.

https://github.com/aws/sagemaker-python-sdk/issues/3319