Fix test_streamlit_deploy_prune_flag database context mismatch

sfc-gh-svishnu · sfc-gh-svishnu · commit 2b9a95457d08 · 2025-11-05T19:47:05.000-08:00
diff --git a/tests_integration/test_streamlit_old.py b/tests_integration/test_streamlit_old.py
@@ -116,7 +116,7 @@ def test_streamlit_deploy_prune_flag(runner, test_database, project_directory):
     # Test prune functionality with legacy (ROOT_LOCATION) deployment
     # Versioned stages are managed differently and don't support manual file uploads well
     streamlit_identifier = "TEST_STREAMLIT_DEPLOY_SNOWCLI"
-    stage_name = f"{test_database}.public.streamlit"
+    stage_name = "streamlit"
     stage_path = f"@{stage_name}/test_streamlit_deploy_snowcli"
 
     def _assert_file_names_on_stage(expected_files: List[str]) -> None:
