Public/Chess_Probabilistic_Programming
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed limit for external engines and added arguments to web.py

Browse Source
This commit is contained in:
Lukas Kuess
2024-01-30 23:07:00 +01:00
parent f3d82d6b19
commit e2ddc3868f
3 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
chesspp/engine.py
View File

@@ -1,6 +1,5 @@
import random
import time
import os
from abc import ABC, abstractmethod
from torch import distributions as dist
@@ -48,6 +47,12 @@ class Limit:
while (time.perf_counter_ns() - start) / 1e9 < self.time:
func(*args, **kwargs)
def translate_to_engine_limit(self) -> chess.engine.Limit:
if self.nodes:
return chess.engine.Limit(nodes=self.nodes)
elif self.time:
return chess.engine.Limit(time=self.time)
class Engine(ABC):
board: chess.Board
@@ -147,7 +152,7 @@ class StockFishEngine(Engine):
self.stockfish = chess.engine.SimpleEngine.popen_uci(path)
def play(self, board: chess.Board, limit: Limit) -> chess.engine.PlayResult:
return self.stockfish.play(board, limit)
return self.stockfish.play(board, limit.translate_to_engine_limit())
@staticmethod
def get_name() -> str:
@@ -160,7 +165,7 @@ class Lc0Engine(Engine):
self.lc0 = chess.engine.SimpleEngine.popen_uci(path)
def play(self, board: chess.Board, limit: Limit) -> chess.engine.PlayResult:
return self.lc0.play(board, limit)
return self.lc0.play(board, limit.translate_to_engine_limit())
@staticmethod
def get_name() -> str: