Continue improvement

Author: mollyheamazon

tests/unit/sagemaker/feature_store/test_feature_group.py

@@ -68,7 +68,7 @@ def s3_uri():
 @pytest.fixture
 def sagemaker_session_mock():
     sagemaker_session_mock = Mock()
-    sagemaker_session_mock.sagemaker_config = {}
+    sagemaker_session_mock.sagemaker_config = None
     return sagemaker_session_mock
 
 

tests/unit/sagemaker/feature_store/test_feature_store.py

@@ -33,7 +33,9 @@
 
 @pytest.fixture
 def sagemaker_session_mock():
-    return Mock()
+    mock = Mock()
+    mock.sagemaker_config = None
+    return mock
 
 
 @pytest.fixture

tests/unit/sagemaker/feature_store/test_feature_utils.py

@@ -42,7 +42,9 @@ def __reduce__(self):
 @pytest.fixture
 def sagemaker_session_mock():
     """Fixture Mock class"""
-    return Mock()
+    mock = Mock()
+    mock.sagemaker_config = None
+    return mock
 
 
 def test_convert_unsupported_types_to_supported(sagemaker_session_mock):

tests/unit/sagemaker/workflow/test_callback_step.py

@@ -28,6 +28,7 @@
 def sagemaker_session_mock():
     session_mock = Mock()
     session_mock.boto_session.client = Mock()
+    session_mock.sagemaker_config = None
     return session_mock
 
 

tox.ini

@@ -94,6 +94,7 @@ commands =
 
     pytest {posargs}
 deps =
+    setuptools
     .[test]
     asyncio
     nest_asyncio