use botocore instead of boto3 to prevent version conflit between aiobotocore and boto3

Author: MalteKurz

doubleml_serverless/double_ml_data_aws.py

@@ -1,6 +1,6 @@
 from doubleml import DoubleMLData
 import os
-import boto3
+from botocore.session import Session
 import pandas as pd
 
 
@@ -50,7 +50,8 @@ def from_s3(cls,
                 x_cols=None,
                 z_cols=None,
                 use_other_treat_as_covariate=True):
-        s3_client = boto3.client('s3')
+        session = Session()
+        s3_client = session.create_client('s3')
         response = s3_client.get_object(Bucket=bucket,
                                         Key=file_key)
         file = response["Body"]
@@ -65,7 +66,8 @@ def store_and_upload_to_s3(self):
         # load csv as a pd.DataFrame
         file_name = os.path.split(self.file_key)[1]
         self.data.to_csv(file_name)
-        s3_client = boto3.client('s3')
+        session = Session()
+        s3_client = session.create_client('s3')
         response = s3_client.upload_file(Filename=file_name,
                                          Bucket=self.bucket,
                                          Key=self.file_key)