Skip to content

Commit 7d91ea1

Browse files
committed
Adding liskov support
1 parent b312460 commit 7d91ea1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

axelrod/strategies/attention.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
CLS_TOKEN = 0
1717
PAD_TOKEN = 1
1818

19-
DEVICES = torch.device("cuda" if torch.cuda.is_available() else "cpu")
19+
DEVICES = torch.device("cpu")
2020

2121
model_weights = load_attention_model_weights()
2222

@@ -329,7 +329,7 @@ def forward(self, input_ids: torch.Tensor) -> torch.Tensor:
329329
pooled_output = self.pooler(sequence_output)
330330
return self.action(pooled_output)
331331

332-
def __eq__(self, other: "PlayerModel") -> bool:
332+
def __eq__(self, other: object) -> bool:
333333
return isinstance(other, PlayerModel)
334334

335335

0 commit comments

Comments
 (0)