fix: update the contribution analysis tool to use original write mode

PiperOrigin-RevId: 828126582
This commit is contained in:
Google Team Member
2025-11-04 14:04:56 -08:00
committed by Copybara-Service
parent 6e5c0eb6e0
commit 54db3d4434
+1 -1
View File
@@ -1057,7 +1057,7 @@ def analyze_contribution(
# Create a session and run the create model query.
original_write_mode = settings.write_mode
try:
if settings.write_mode == WriteMode.BLOCKED:
if original_write_mode == WriteMode.BLOCKED:
raise ValueError("analyze_contribution is not allowed in this session.")
elif original_write_mode != WriteMode.PROTECTED:
# Running create temp model requires a session. So we set the write mode