Update max retry and fix regex for tf metric

Author: Dan Choi

tests/conftest.py

@@ -16,6 +16,8 @@
 
 import boto3
 import pytest
+from botocore.config import Config
+from botocore.regions import EndpointResolver
 
 from sagemaker import Session
 from sagemaker.local import LocalSession
@@ -32,7 +34,7 @@ def pytest_addoption(parser):
 @pytest.fixture(scope='session')
 def sagemaker_client_config(request):
     config = request.config.getoption('--sagemaker-client-config')
-    return json.loads(config) if config else None
+    return json.loads(config) if config else dict()
 
 
 @pytest.fixture(scope='session')
@@ -50,6 +52,9 @@ def boto_config(request):
 @pytest.fixture(scope='session')
 def sagemaker_session(sagemaker_client_config, sagemaker_runtime_config, boto_config):
     boto_session = boto3.Session(**boto_config) if boto_config else boto3.Session(region_name=DEFAULT_REGION)
+
+    sagemaker_client_config.setdefault('config', Config(retries=dict(max_attempts=10)))
+
     sagemaker_client = boto_session.client('sagemaker', **sagemaker_client_config) if sagemaker_client_config else None
     runtime_client = (boto_session.client('sagemaker-runtime', **sagemaker_runtime_config) if sagemaker_runtime_config
                       else None)

tests/integ/test_tuner.py

@@ -232,7 +232,7 @@ def test_tuning_tf(sagemaker_session):
         hyperparameter_ranges = {'learning_rate': ContinuousParameter(0.05, 0.2)}
 
         objective_metric_name = 'loss'
-        metric_definitions = [{'Name': 'loss', 'Regex': 'loss=([0-9\\.]+)'}]
+        metric_definitions = [{'Name': 'loss', 'Regex': 'loss = ([0-9\\.]+)'}]
 
         tuner = HyperparameterTuner(estimator, objective_metric_name, hyperparameter_ranges, metric_definitions,
                                     objective_type='Minimize', max_jobs=2, max_parallel_jobs=2)