rlcard-showdown/server/tournament/rlcard_wrap/__init__.py

27 lines
693 B
Python
Raw Normal View History

2020-05-11 13:36:49 +08:00
import rlcard
from .leduc_holdem_random_model import LeducHoldemRandomModelSpec
2020-05-12 05:42:08 +08:00
from .doudizhu_random_model import DoudizhuRandomModelSpec
2020-05-11 13:36:49 +08:00
# Register Leduc Holdem Random Model
rlcard.models.registration.model_registry.model_specs['leduc-holdem-random'] = LeducHoldemRandomModelSpec()
2020-05-12 05:42:08 +08:00
# Register Doudizhu Random Model
rlcard.models.registration.model_registry.model_specs['doudizhu-random'] = DoudizhuRandomModelSpec()
2020-05-11 13:36:49 +08:00
# The models we are concerned
MODEL_IDS = {}
MODEL_IDS['leduc-holdem'] = [
'leduc-holdem-random',
'leduc-holdem-cfr',
'leduc-holdem-rule-v1',
]
2020-05-12 05:42:08 +08:00
MODEL_IDS['doudizhu'] = [
'doudizhu-random',
2020-05-13 07:51:16 +08:00
'doudizhu-rule-v1',
2020-05-12 05:42:08 +08:00
]
2020-05-13 10:47:43 +08:00