We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c94bdce commit 5f01f72Copy full SHA for 5f01f72
1 file changed
src/inference/service.py
@@ -12,10 +12,10 @@
12
from game.actions import ACTION_SPACE
13
from game.board import AtaxxBoard
14
from game.types import Move
15
-from model.system import AtaxxZero
16
17
if TYPE_CHECKING:
18
from engine.mcts import MCTS
+ from model.system import AtaxxZero
19
20
InferenceMode = Literal["fast", "strong"]
21
@@ -133,6 +133,8 @@ def _require_torch() -> ModuleType:
133
return torch_module
134
135
def _load_system(self) -> AtaxxZero:
136
137
+
138
torch_module = self._require_torch()
139
ckpt = self.checkpoint_path
140
if ckpt.suffix == ".ckpt":
0 commit comments