Commitdc3a1fb

committed

move use_sea init to Session constructor

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

1 parentbb3e94d commitdc3a1fbCopy full SHA for dc3a1fb

File tree

-3

lines changed

-3

lines changed

Lines changed: 4 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -76,7 +76,9 @@ def __init__(`
`76`	`76`	`tls_client_cert_key_password=kwargs.get("_tls_client_cert_key_password"),`
`77`	`77`	`)`
`78`	`78`
	`79`	`+self.use_sea=kwargs.get("use_sea",False)`
`79`	`80`	`self.backend=self._create_backend(`
	`81`	`+self.use_sea,`
`80`	`82`	`server_hostname,`
`81`	`83`	`http_path,`
`82`	`84`	`all_headers,`
`@@ -89,6 +91,7 @@ def __init__(`
`89`	`91`
`90`	`92`	`def_create_backend(`
`91`	`93`	`self,`
	`94`	`+use_sea:bool,`
`92`	`95`	`server_hostname:str,`
`93`	`96`	`http_path:str,`
`94`	`97`	`all_headers:List[Tuple[str,str]],`
`@@ -97,10 +100,8 @@ def _create_backend(`
`97`	`100`	`kwargs:dict,`
`98`	`101`	`)->DatabricksClient:`
`99`	`102`	`"""Create and return the appropriate backend client."""`
`100`		`-self.use_sea=kwargs.get("use_sea",False)`
`101`		`-`
`102`	`103`	`databricks_client_class:Type[DatabricksClient]`
`103`		`-ifself.use_sea:`
	`104`	`+ifuse_sea:`
`104`	`105`	`logger.debug("Creating SEA backend client")`
`105`	`106`	`databricks_client_class=SeaDatabricksClient`
`106`	`107`	`else:`

Comments

(0)