Fix credentials provider configuration

laughingman7743 · Jan 10, 2017 · fec9452 · fec9452
1 parent e92d7b1
commit fec9452
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyathenajdbc/connection.py b/pyathenajdbc/connection.py
@@ -89,14 +89,14 @@ def _build_driver_args(self, **kwargs):
         props = jpype.java.util.Properties()
         if self.credential_file:
             props.setProperty('aws_credentials_provider_class',
+                              'com.amazonaws.athena.jdbc.shaded.' +
                               'com.amazonaws.auth.PropertiesFileCredentialsProvider')
             props.setProperty('aws_credentials_provider_arguments',
                               self.credential_file)
-        if self.token:
+        elif self.token:
             props.setProperty('aws_credentials_provider_class',
-                              'com.amazonaws.athena.jdbc.CustomSessionCredentialsProvider')
-            props.setProperty('aws_credentials_provider_arguments',
-                              ','.join([self.access_key, self.secret_key, self.token]))
+                              'com.amazonaws.athena.jdbc.shaded.' +
+                              'com.amazonaws.auth.InstanceProfileCredentialsProvider')
         else:
             props.setProperty('user', self.access_key)
             props.setProperty('password', self.secret_key)