bettermint commited on
Commit
d828cf7
·
verified ·
1 Parent(s): 45219af

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -30,7 +30,7 @@ class EngineChess:
30
  self.thread.start()
31
 
32
  self._has_quit_command_been_sent = False
33
- self._debug_view = False
34
 
35
  def _put(self, command):
36
  if not self._stockfish.stdin:
@@ -55,7 +55,7 @@ class EngineChess:
55
  return ""
56
 
57
  if self._debug_view:
58
- print(line.strip())
59
  return line.strip()
60
 
61
  def _is_ready(self) -> None:
@@ -83,7 +83,7 @@ app.add_middleware(
83
  async def websocket_endpoint(websocket: WebSocket, version: str):
84
  await websocket.accept()
85
 
86
- stockfish = EngineChess(f"engines/stockfish/stockfish-{version}-uci")
87
 
88
  async def read_from_socket(websocket: WebSocket):
89
  async for data in websocket.iter_text():
@@ -105,7 +105,7 @@ async def websocket_endpoint(websocket: WebSocket, version: str):
105
  async def websocket_endpoint(websocket: WebSocket, elo: str):
106
  await websocket.accept()
107
 
108
- stockfish = EngineChess(["./engines/maia/lc0", f"--weights=engines/maia-{elo}.pb.gz", "--backend=trivial"])
109
 
110
  async def read_from_socket(websocket: WebSocket):
111
  async for data in websocket.iter_text():
 
30
  self.thread.start()
31
 
32
  self._has_quit_command_been_sent = False
33
+ self._debug_view = True
34
 
35
  def _put(self, command):
36
  if not self._stockfish.stdin:
 
55
  return ""
56
 
57
  if self._debug_view:
58
+ print(f"Engine {path_engine[0]}:", line.strip())
59
  return line.strip()
60
 
61
  def _is_ready(self) -> None:
 
83
  async def websocket_endpoint(websocket: WebSocket, version: str):
84
  await websocket.accept()
85
 
86
+ stockfish = EngineChess([f"engines/stockfish/stockfish-{version}-uci]")
87
 
88
  async def read_from_socket(websocket: WebSocket):
89
  async for data in websocket.iter_text():
 
105
  async def websocket_endpoint(websocket: WebSocket, elo: str):
106
  await websocket.accept()
107
 
108
+ stockfish = EngineChess(["./engines/maia/lc0", "--backend=trivial", f"--weights=engines/maia-{elo}.pb.gz"])
109
 
110
  async def read_from_socket(websocket: WebSocket):
111
  async for data in websocket.iter_text():