fix: Extracted profile_name directly from sagemaker.Session if None

Author: Kusumlata Bhatt

src/sagemaker/feature_store/feature_group.py

@@ -804,6 +804,9 @@ def ingest(
         if max_workers <= 0:
             raise RuntimeError("max_workers must be greater than 0.")
 
+        if profile_name is None:
+            profile_name = self.sagemaker_session.boto_session.profile_name
+
         manager = IngestionManagerPandas(
             feature_group_name=self.name,
             sagemaker_session=self.sagemaker_session,

tests/unit/sagemaker/feature_store/test_feature_group.py

@@ -311,7 +311,7 @@ def test_ingest(ingestion_manager_init, sagemaker_session_mock, fs_runtime_clien
         sagemaker_fs_runtime_client_config=fs_runtime_client_config_mock,
         max_workers=10,
         max_processes=1,
-        profile_name=None,
+        profile_name=sagemaker_session_mock.boto_session.profile_name,
     )
     mock_ingestion_manager_instance.run.assert_called_once_with(
         data_frame=df, wait=True, timeout=None
@@ -337,7 +337,7 @@ def test_ingest_default(ingestion_manager_init, sagemaker_session_mock):
         sagemaker_fs_runtime_client_config=fs_runtime_client_config_mock,
         max_workers=1,
         max_processes=1,
-        profile_name=None,
+        profile_name=sagemaker_session_mock.boto_session.profile_name,
     )
     mock_ingestion_manager_instance.run.assert_called_once_with(
         data_frame=df, wait=True, timeout=None