Update record_writer.py (#521)

lanpa · Nov 4, 2019 · ccccddf · ccccddf
1 parent 656037e
commit ccccddf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorboardX/record_writer.py b/tensorboardX/record_writer.py
@@ -3,6 +3,7 @@
 The code was borrowed from https://github.com/TeamHG-Memex/tensorboard_logger
 """
 
+import os
 import copy
 import io
 import os.path
@@ -79,7 +80,7 @@ def write(self, val):
         self.buffer.write(val)
 
     def flush(self):
-        s3 = boto3.client('s3')
+        s3 = boto3.client('s3', endpoint_url=os.environ.get('S3_ENDPOINT'))
         bucket, path = self.bucket_and_path()
         upload_buffer = copy.copy(self.buffer)
         upload_buffer.seek(0)