DouZero
/
rlcard-showdown
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix delete model bug

This commit is contained in:
Daochen Zha 2020-12-06 16:27:24 -06:00
parent e63f4626b4
commit d6cc4dbe7d
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server/tournament/views.py
View File

@ -19,9 +19,9 @@ from .models import Game, Payoff, UploadedAgent
from .tournament import Tournament from .tournament import Tournament
from .rlcard_wrap import rlcard, MODEL_IDS from .rlcard_wrap import rlcard, MODEL_IDS
MODEL_IDS_ALL = copy.deepcopy(MODEL_IDS)
def _reset_model_ids(): def _get_model_ids_all():
MODEL_IDS_ALL = copy.deepcopy(MODEL_IDS)
agents = UploadedAgent.objects.all() agents = UploadedAgent.objects.all()
for agent in agents: for agent in agents:
path = os.path.join(settings.MEDIA_ROOT, agent.f.name) path = os.path.join(settings.MEDIA_ROOT, agent.f.name)
@ -44,7 +44,7 @@ def _reset_model_ids():
return model return model
rlcard.models.registration.model_registry.model_specs[name] = ModelSpec() rlcard.models.registration.model_registry.model_specs[name] = ModelSpec()
MODEL_IDS_ALL[game].append(name) MODEL_IDS_ALL[game].append(name)
_reset_model_ids() return MODEL_IDS_ALL
PAGE_FIELDS = ['elements_every_page', 'page_index'] PAGE_FIELDS = ['elements_every_page', 'page_index']
@ -105,6 +105,7 @@ def launch(request):
except: except:
return HttpResponse(json.dumps({'value': -1, 'info': 'parameters error'})) return HttpResponse(json.dumps({'value': -1, 'info': 'parameters error'}))
MODEL_IDS_ALL = _get_model_ids_all()
games_data, payoffs_data = Tournament(game, MODEL_IDS_ALL[game], eval_num).launch() games_data, payoffs_data = Tournament(game, MODEL_IDS_ALL[game], eval_num).launch()
Game.objects.filter(name=game).delete() Game.objects.filter(name=game).delete()
Payoff.objects.filter(name=game).delete() Payoff.objects.filter(name=game).delete()
@ -137,7 +138,6 @@ def upload_agent(request):
a = UploadedAgent(name=name, game=game, f=f, entry=entry) a = UploadedAgent(name=name, game=game, f=f, entry=entry)
a.save() a.save()
_reset_model_ids()
return HttpResponse(json.dumps({'value': 0, 'info': 'success'})) return HttpResponse(json.dumps({'value': 0, 'info': 'success'}))
def delete_agent(request): def delete_agent(request):
@ -149,7 +149,6 @@ def delete_agent(request):
UploadedAgent.objects.filter(name=name).delete() UploadedAgent.objects.filter(name=name).delete()
Game.objects.filter(agent0=name).delete() Game.objects.filter(agent0=name).delete()
Game.objects.filter(agent1=name).delete() Game.objects.filter(agent1=name).delete()
_reset_model_ids()
return HttpResponse(json.dumps({'value': 0, 'info': 'success'})) return HttpResponse(json.dumps({'value': 0, 'info': 'success'}))
def list_uploaded_agents(request): def list_uploaded_agents(request):