Skip to content

Commit 02fb608

Browse files
committed
[db] find_one_params with try
1 parent bb50d2a commit 02fb608

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

tensorlayer/db.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -165,10 +165,13 @@ def find_one_params(self, args={}):
165165
return False, False
166166
# print(f_id)
167167
# exit()
168-
params = pickle.loads(self.paramsfs.get(f_id).read())
169-
# print(self.paramsfs)
170-
print("[TensorDB] Find one params SUCCESS, {} took: {}s".format(args, round(time.time()-s, 2)))
171-
return params, f_id
168+
try:
169+
params = pickle.loads(self.paramsfs.get(f_id).read())
170+
# print(self.paramsfs)
171+
print("[TensorDB] Find one params SUCCESS, {} took: {}s".format(args, round(time.time()-s, 2)))
172+
return params, f_id
173+
except:
174+
return False, False
172175

173176
def find_all_params(self, args={}):
174177
""" Find all parameter from MongoDB Buckets """

0 commit comments

Comments
 (0)