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

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -8,6 +8,8 @@ import os
8
  from threading import Thread
9
  from queue import Queue, Empty
10
 
 
 
11
  def enqueue_output(out, queue):
12
  for line in iter(out.readline, b''):
13
  queue.put(line)
@@ -103,7 +105,7 @@ async def websocket_endpoint(websocket: WebSocket, version: str):
103
  async def websocket_endpoint(websocket: WebSocket, elo: str):
104
  await websocket.accept()
105
 
106
- stockfish = EngineChess(["export", "LD_LIBRARY_PATH=/engines/maia:$LD_LIBRARY_PATH", "&&", "./engines/maia/lc0", f"--weights=engines/maia-{elo}.pb.gz", "--backend=trivial"])
107
 
108
  async def read_from_socket(websocket: WebSocket):
109
  async for data in websocket.iter_text():
 
8
  from threading import Thread
9
  from queue import Queue, Empty
10
 
11
+ os.environ['LD_LIBRARY_PATH'] = '/engines/maia:' + os.environ.get('LD_LIBRARY_PATH', '')
12
+
13
  def enqueue_output(out, queue):
14
  for line in iter(out.readline, b''):
15
  queue.put(line)
 
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():