great work, having troubles running with minivilles

[tal-tepper]

Hi,

thanks for this repo, it looks amazing.
I tried to play/train minivilles and it crushes during MCTS.
it looks numba related, I tried to add NUMBA_DISABLE_JIT=1 in pit for playing and it didn't help(probably did it wrong).
part of the error:
File "/home/user/repos/alpha-zero-general/MCTS.py", line 164, in search
a, next_s, next_player = get_next_best_action_and_canonical_state(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/user/repos/alpha-zero-general/alpha_env/lib/python3.11/site-packages/numba/core/dispatcher.py", line 423, in _compile_for_args
error_rewrite(e, 'typing')
File "/home/user/repos/alpha-zero-general/alpha_env/lib/python3.11/site-packages/numba/core/dispatcher.py", line 364, in error_rewrite
raise e.with_traceback(None)
numba.core.errors.TypingError: Failed in nopython mode pipeline (step: nopython frontend)

• Resolution failure for literal arguments:
  Failed in nopython mode pipeline (step: nopython frontend)
• Resolution failure for literal arguments:
  Failed in nopython mode pipeline (step: nopython frontend)
• Resolution failure for literal arguments:
  Failed in nopython mode pipeline (step: nopython frontend)
  No implementation of function Function(<function clip at 0x75d085a94220>) found for signature:

clip(int64, Literalint, Literalint)

thank you in advance
if you don't have time and just throw me a lead it will also be great

[cestpasphoto]

Hi @tal-tepper ,

No issue without Numba:
NUMBA_DISABLE_JIT=1 python3 ./pit.py minivilles minivilles/pretrained_2players.pt minivilles/pretrained_2players.pt -n 1 --display

but Numba seems not to support np.clip properly as you highlighted. So I just rewrote it and it seems to work, please check main branch with latest commit. Note that I didn't expected anyone to use this game except me so there's many things written in French on the console during play ...