@@ -85,13 +85,16 @@ def __init__(self, ip='localhost', port=27017, db_name='db_name', user_name=None
8585 self .db_name = db_name
8686 self .user_name = user_name
8787
88+ @classmethod
8889 def __autofill (self , args ):
8990 return args .update ({'studyID' : self .studyID })
9091
91- def __serialization (self , ps ):
92+ @staticmethod
93+ def __serialization (ps ):
9294 return pickle .dumps (ps , protocol = 2 )
9395
94- def __deserialization (self , ps ):
96+ @staticmethod
97+ def __deserialization (ps ):
9598 return pickle .loads (ps )
9699
97100 def save_params (self , params = None , args = None ): #, file_name='parameters'):
@@ -298,20 +301,25 @@ def test_log(self, args=None):
298301 return _result
299302
300303 @AutoFill
301- def del_test_log (self , args = {} ):
304+ def del_test_log (self , args = None ):
302305 """ Delete test log.
303306
304307 Parameters
305308 -----------
306309 args : dictionary, find items to delete, leave it empty to delete all log.
307310 """
311+ if args is None :
312+ args = {}
308313
309314 self .db .TestLog .delete_many (args )
310315 print ("[TensorDB] Delete TestLog SUCCESS" )
311316
312- ## =========================== Network Architecture ================== ##
317+ # =========================== Network Architecture ================== ##
313318 @AutoFill
314- def save_model_architecture (self , s , args = {}):
319+ def save_model_architecture (self , s , args = None ):
320+ if args is None :
321+ args = {}
322+
315323 self .__autofill (args )
316324 fid = self .archfs .put (s , filename = "modelarchitecture" )
317325 args .update ({"fid" : fid })
0 commit comments