Fix unit tests failure

Author: mollyheamazon

requirements/extras/test_requirements.txt

@@ -1,4 +1,5 @@
 tox==3.24.5
+setuptools
 numpy>=2.0.0, <3.0
 build[virtualenv]==1.2.1
 flake8==7.1.2

tests/unit/sagemaker/feature_store/test_dataset_builder.py

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