Update way of setting sessions_path.

Author: zhaozuy

python/model_hosting_container_standards/sagemaker/sessions/manager.py

@@ -139,7 +139,7 @@ def __init__(self, properties: dict):
         else:
             session_dir = os.path.join(tempfile.gettempdir(), "sagemaker_sessions")
 
-        self.sessions_path = properties.get("sessions_path", session_dir)
+        self.sessions_path = properties.get("sessions_path") or session_dir
         self.sessions: dict[str, Session] = {}
         self._lock = RLock()  # Thread safety for concurrent session access
 

python/model_hosting_container_standards/sagemaker/sessions/transform.py

@@ -45,7 +45,9 @@ def _parse_session_request(request_data: dict) -> Optional[SessionRequest]:
         return None
 
 
-def _validate_session_if_present(raw_request: Request, session_manager: SessionManager):
+def _validate_session_if_present(
+    raw_request: Request, session_manager: Optional[SessionManager]
+):
     """Validate that the session ID in the request exists and is not expired.
 
     Args:

python/model_hosting_container_standards/sagemaker/sessions/utils.py

@@ -1,5 +1,6 @@
 import logging
 from http import HTTPStatus
+from typing import Optional
 
 from fastapi import Request
 from fastapi.exceptions import HTTPException
@@ -24,7 +25,7 @@ def get_session_id_from_request(raw_request: Request):
     return raw_request.headers.get(SageMakerSessionHeader.SESSION_ID)
 
 
-def get_session(session_manager: SessionManager, raw_request: Request):
+def get_session(session_manager: Optional[SessionManager], raw_request: Request):
     """Retrieve the session associated with the request.
 
     Args: