diff --git a/mjai/bot/key.txt b/mjai/bot/key.txt deleted file mode 100644 index 454f6b3..0000000 --- a/mjai/bot/key.txt +++ /dev/null @@ -1 +0,0 @@ -0123456789abcdef \ No newline at end of file diff --git a/mjai/bot/model.py b/mjai/bot/model.py index 214227b..1d65f24 100644 --- a/mjai/bot/model.py +++ b/mjai/bot/model.py @@ -250,8 +250,6 @@ def __init__( boltzmann_epsilon = 0, boltzmann_temp = 1, top_p = 1, - r = False, - key = "", ): self.engine_type = 'mortal' self.device = device or torch.device('cpu') @@ -270,8 +268,6 @@ def __init__( self.boltzmann_epsilon = boltzmann_epsilon self.boltzmann_temp = boltzmann_temp self.top_p = top_p - self.r = r - self.key = key def react_batch(self, obs, masks, invisible_obs): with ( @@ -340,15 +336,6 @@ def load_model(seat: int) -> riichi.mjai.Bot: control_state_file = pathlib.Path(__file__).parent / control_state_file state = torch.load(control_state_file, map_location=device) - key = "" - if 'r' in state['config']: - del state - from . import obfuscated_model - engine = obfuscated_model.IIIIIllllll() - return riichi.mjai.Bot(engine, seat) - else: - r = False - mortal = Brain(version=state['config']['control']['version'], conv_channels=state['config']['resnet']['conv_channels'], num_blocks=state['config']['resnet']['num_blocks']).eval() dqn = DQN(version=state['config']['control']['version']).eval() mortal.load_state_dict(state['mortal']) @@ -364,8 +351,6 @@ def load_model(seat: int) -> riichi.mjai.Bot: enable_rule_based_agari_guard = False, name = 'mortal', version = state['config']['control']['version'], - r = r, - key = key, ) bot = riichi.mjai.Bot(engine, seat)