add HuggingFaceModel and HuggingFaceEstimator to __init__.py

Wang Napieralski · Wang Napieralski · commit 6158dd17fabe · 2021-06-17T09:23:17.000-07:00
diff --git a/src/sagemaker/huggingface/__init__.py b/src/sagemaker/huggingface/__init__.py
@@ -14,3 +14,4 @@
 from __future__ import absolute_import
 
 from sagemaker.huggingface.estimator import HuggingFace  # noqa: F401
+from sagemaker.huggingface.model import HuggingFaceModel, HuggingFacePredictor  # noqa: F401

Original file line number	Diff line number	Diff line change
`@@ -14,3 +14,4 @@`
`14`	`14`	`from __future__ import absolute_import`
`15`	`15`
`16`	`16`	`from sagemaker.huggingface.estimator import HuggingFace # noqa: F401`
	`17`	`+from sagemaker.huggingface.model import HuggingFaceModel, HuggingFacePredictor # noqa: F401`