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

Changing exception type

Browse Source
This commit is contained in:
Ajay Deshpande
2018-10-05 15:57:50 -07:00
committed by zach dwiel
parent a7f5442015
commit 5eac0102de
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rl_coach/data_stores/s3_data_store.py
View File

@@ -5,7 +5,7 @@ from minio.error import ResponseError
from configparser import ConfigParser, Error from configparser import ConfigParser, Error
from google.protobuf import text_format from google.protobuf import text_format
from tensorflow.python.training.checkpoint_state_pb2 import CheckpointState from tensorflow.python.training.checkpoint_state_pb2 import CheckpointState
from minio.error import ResponseError
import os import os
import time import time
import io import io
@@ -71,6 +71,8 @@ class S3DataStore(DataStore):
abs_name = os.path.abspath(os.path.join(checkpoint_file[0], checkpoint_file[1])) abs_name = os.path.abspath(os.path.join(checkpoint_file[0], checkpoint_file[1]))
rel_name = os.path.relpath(abs_name, self.params.checkpoint_dir) rel_name = os.path.relpath(abs_name, self.params.checkpoint_dir)
self.mc.fput_object(self.params.bucket_name, rel_name, abs_name)
print("Deleting lock file") print("Deleting lock file")
self.mc.remove_object(self.params.bucket_name, self.params.lock_file) self.mc.remove_object(self.params.bucket_name, self.params.lock_file)
@@ -88,7 +90,7 @@ class S3DataStore(DataStore):
if next(objects, None) is None: if next(objects, None) is None:
try: try:
self.mc.fget_object(self.params.bucket_name, "checkpoint", filename) self.mc.fget_object(self.params.bucket_name, "checkpoint", filename)
except ResponseError as e: except Exception as e:
continue continue
break break
time.sleep(10) time.sleep(10)