fix client BallistaContext

Author: Wang

datafusion-cli/src/context.rs

@@ -33,7 +33,7 @@ pub enum Context {
 impl Context {
     /// create a new remote context with given host and port
     pub async fn new_remote(host: &str, port: u16) -> Result<Context> {
-        Ok(Context::Remote(BallistaContext::try_new(host, port)?))
+        Ok(Context::Remote(BallistaContext::try_new(host, port).await?))
     }
 
     /// create a local context using the given config
@@ -61,7 +61,7 @@ impl BallistaContext {
         use ballista::prelude::BallistaConfig;
         let config: BallistaConfig = BallistaConfig::new()
             .map_err(|e| DataFusionError::Execution(format!("{:?}", e)))?;
-        BallistaContext::remote(host, port, &config).await
+        Ok(Self(BallistaContext::remote(host, port, &config).await?))
     }
 
     pub async fn sql(&mut self, sql: &str) -> Result<Arc<dyn DataFrame>> {