33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import sys, json, os, traceback
|
|
sys.path.insert(0, r'F:\projects\gogo2')
|
|
res={}
|
|
try:
|
|
from utils.database_manager import get_database_manager
|
|
db=get_database_manager()
|
|
def db_count(name):
|
|
try:
|
|
lst = db.list_checkpoints(name)
|
|
return len(lst) if lst is not None else 0
|
|
except Exception as e:
|
|
print("DB error for %s: %s" % (name, str(e)))
|
|
return -1
|
|
res.setdefault('db', {})['dqn_agent']=db_count('dqn_agent')
|
|
res['db']['enhanced_cnn']=db_count('enhanced_cnn')
|
|
except Exception as e:
|
|
res['db']={'error': str(e)}
|
|
try:
|
|
from utils.checkpoint_manager import get_checkpoint_manager
|
|
cm=get_checkpoint_manager()
|
|
def fs_count(name):
|
|
try:
|
|
lst = cm.get_all_checkpoints(name)
|
|
return len(lst) if lst is not None else 0
|
|
except Exception as e:
|
|
print("FS error for %s: %s" % (name, str(e)))
|
|
return -1
|
|
res.setdefault('fs', {})['dqn_agent']=fs_count('dqn_agent')
|
|
res['fs']['enhanced_cnn']=fs_count('enhanced_cnn')
|
|
except Exception as e:
|
|
res['fs']={'error': str(e)}
|
|
print(json.dumps(res))
|