fix: check for distributions in TF estimator

ajaykarpur · ajaykarpur · commit 0c64068562e5 · 2020-12-02T11:33:35.000-08:00
diff --git a/src/sagemaker/tensorflow/estimator.py b/src/sagemaker/tensorflow/estimator.py
@@ -113,6 +113,7 @@ def __init__(
             :class:`~sagemaker.estimator.Framework` and
             :class:`~sagemaker.estimator.EstimatorBase`.
         """
+        distribution = renamed_kwargs("distributions", "distribution", distribution, kwargs)
         instance_type = renamed_kwargs(
             "train_instance_type", "instance_type", kwargs.get("instance_type"), kwargs
         )

Original file line number	Diff line number	Diff line change
`@@ -113,6 +113,7 @@ def __init__(`
`113`	`113`	:class:`~sagemaker.estimator.Framework` and
`114`	`114`	:class:`~sagemaker.estimator.EstimatorBase`.
`115`	`115`	`"""`
	`116`	`+ distribution = renamed_kwargs("distributions", "distribution", distribution, kwargs)`
`116`	`117`	`instance_type = renamed_kwargs(`
`117`	`118`	`"train_instance_type", "instance_type", kwargs.get("instance_type"), kwargs`
`118`	`119`	`)`