gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

Fix breaking change from minio update (#469)

Browse Source 
`ResponseError` replaced by `S3Error` in new minio version
This commit is contained in:
Guy Jacob
2020-12-15 10:02:16 +02:00
committed by GitHub
parent 59e08034c6
commit f52ff1784d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rl_coach/data_stores/s3_data_store.py
View File

@@ -18,7 +18,7 @@
from rl_coach.data_stores.data_store import DataStoreParameters from rl_coach.data_stores.data_store import DataStoreParameters
from rl_coach.data_stores.checkpoint_data_store import CheckpointDataStore from rl_coach.data_stores.checkpoint_data_store import CheckpointDataStore
from minio import Minio from minio import Minio
from minio.error import ResponseError from minio.error import S3Error
from configparser import ConfigParser, Error from configparser import ConfigParser, Error
from rl_coach.checkpoint import CheckpointStateFile from rl_coach.checkpoint import CheckpointStateFile
from rl_coach.data_stores.data_store import SyncFiles from rl_coach.data_stores.data_store import SyncFiles
@@ -134,7 +134,7 @@ class S3DataStore(CheckpointDataStore):
for filename in os.listdir(os.path.join(self.params.expt_dir, 'gifs')): for filename in os.listdir(os.path.join(self.params.expt_dir, 'gifs')):
self.mc.fput_object(self.params.bucket_name, filename, os.path.join(self.params.expt_dir, 'gifs', filename)) self.mc.fput_object(self.params.bucket_name, filename, os.path.join(self.params.expt_dir, 'gifs', filename))
except ResponseError as e: except S3Error as e:
print("Got exception: %s\n while saving to S3", e) print("Got exception: %s\n while saving to S3", e)
def load_from_store(self): def load_from_store(self):
@@ -190,7 +190,7 @@ class S3DataStore(CheckpointDataStore):
if not os.path.exists(filename): if not os.path.exists(filename):
self.mc.fget_object(obj.bucket_name, obj.object_name, filename) self.mc.fget_object(obj.bucket_name, obj.object_name, filename)
except ResponseError as e: except S3Error as e:
print("Got exception: %s\n while loading from S3", e) print("Got exception: %s\n while loading from S3", e)
def setup_checkpoint_dir(self, crd=None): def setup_checkpoint_dir(self, crd=None):