Skip to content

Commit 0742bf6

Browse files
committed
fix existing unit test error with sagemaker-train evaluate test execution
1 parent 7d1ec22 commit 0742bf6

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

sagemaker-train/tests/unit/train/evaluate/test_execution.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,10 +57,17 @@ class MockUnassigned:
5757

5858
@pytest.fixture
5959
def mock_session():
60-
"""Mock SageMaker session."""
60+
"""Mock SageMaker session that passes isinstance checks."""
61+
from sagemaker.core.helper.session_helper import Session
62+
63+
# Create a mock that will pass isinstance(obj, Session) checks
6164
session = MagicMock()
6265
session.boto_region_name = DEFAULT_REGION
6366
session.client.return_value = MagicMock()
67+
68+
# Make isinstance check pass
69+
session.__class__ = type('MockSession', (Session,), {})
70+
6471
return session
6572

6673

0 commit comments

Comments
 (0)