Updating PR.

Fixing omissions and formating.

Author: Christian Osendorfer

src/sagemaker/estimator.py

@@ -598,9 +598,11 @@ def _json_encode_hyperparameters(hyperparameters: Dict[str, Any]) -> Dict[str, A
         current_hyperparameters = hyperparameters
         if current_hyperparameters is not None:
             hyperparameters = {
-               str(k): (v if isinstance(
-                    v, (str, int, float, bool, Parameter, Expression, Properties)
-                    ) else json.dumps(v))
+                str(k): (
+                    v
+                    if isinstance(v, (str, int, float, bool, Parameter, Expression, Properties))
+                    else json.dumps(v)
+                )
                 for (k, v) in current_hyperparameters.items()
             }
         return hyperparameters

src/sagemaker/local/image.py

@@ -493,7 +493,7 @@ def _prepare_training_volumes(
         # If there is a training script directory and it is a local directory,
         # mount it to the container.
         if sagemaker.estimator.DIR_PARAM_NAME in hyperparameters:
-            training_dir = json.loads(hyperparameters[sagemaker.estimator.DIR_PARAM_NAME])
+            training_dir = hyperparameters[sagemaker.estimator.DIR_PARAM_NAME]
             parsed_uri = urlparse(training_dir)
             if parsed_uri.scheme == "file":
                 host_dir = os.path.abspath(parsed_uri.netloc + parsed_uri.path)

tests/unit/test_estimator.py

@@ -1337,8 +1337,7 @@ def test_shuffle_config(sagemaker_session):
 
 BASE_HP = {
     "sagemaker_program": SCRIPT_NAME,
-    "sagemaker_submit_directory":
-        "s3://mybucket/{}/source/sourcedir.tar.gz".format(JOB_NAME),
+    "sagemaker_submit_directory": "s3://mybucket/{}/source/sourcedir.tar.gz".format(JOB_NAME),
     "sagemaker_job_name": JOB_NAME,
 }