use proper manager

sfc-gh-daniszewski · sfc-gh-daniszewski · commit ac56f316e174 · 2025-09-25T18:25:06.000+02:00
diff --git a/src/snowflake/cli/_plugins/streamlit/streamlit_entity.py b/src/snowflake/cli/_plugins/streamlit/streamlit_entity.py
@@ -133,7 +133,7 @@ def deploy(
                 self.get_deploy_sql(replace=replace, from_stage_name=stage_root)
             )
 
-            StreamlitManager().grant_privileges(self.model)
+            StreamlitManager(connection=self._conn).grant_privileges(self.model)
 
         return self.perform(EntityActions.GET_URL, action_context, *args, **kwargs)
 
@@ -260,4 +260,4 @@ def _deploy_experimental(
             force_overwrite=True,  # files copied to streamlit vstage need to be overwritten
         )
 
-        StreamlitManager().grant_privileges(self.model)
+        StreamlitManager(connection=self._conn).grant_privileges(self.model)

Original file line number	Diff line number	Diff line change
`@@ -133,7 +133,7 @@ def deploy(`
`133`	`133`	`self.get_deploy_sql(replace=replace, from_stage_name=stage_root)`
`134`	`134`	`)`
`135`	`135`
`136`		`- StreamlitManager().grant_privileges(self.model)`
	`136`	`+ StreamlitManager(connection=self._conn).grant_privileges(self.model)`
`137`	`137`
`138`	`138`	`return self.perform(EntityActions.GET_URL, action_context, args, *kwargs)`
`139`	`139`
`@@ -260,4 +260,4 @@ def _deploy_experimental(`
`260`	`260`	`force_overwrite=True, # files copied to streamlit vstage need to be overwritten`
`261`	`261`	`)`
`262`	`262`
`263`		`- StreamlitManager().grant_privileges(self.model)`
	`263`	`+ StreamlitManager(connection=self._conn).grant_privileges(self.model)`