lib/dynamodb_manager.rb in dynamodb_geo-0.1.6 vs lib/dynamodb_manager.rb in dynamodb_geo-0.1.8

@@ -9,10 +9,10 @@
       access_key_id = ENV['AWS_ACCESS_KEY_ID']
       secret_access_key = ENV['AWS_SECRET_ACCESS_KEY']
 
       credentials = Aws::SharedCredentials.new(profile_name: profile_name).credentials if access_key_id.nil? && secret_access_key.nil?
     end
-    credentials = Aws::Credentials.new(access_key_id, secret_access_key)
+    credentials = Aws::Credentials.new(access_key_id, secret_access_key, ENV['AWS_SESSION_TOKEN'])
 
     @table_name      = table_name
     @hash_key        = 'hashkey'
     @range_key       = 'rangekey'
     @geohash_key     = 'geohash'