Skip to content

Commit a4f275b

Browse files
committed
Removing stochastic strategy in test
1 parent e800b3e commit a4f275b

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

axelrod/load_data_.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,5 +86,7 @@ def load_attention_model_weights(
8686
):
8787
"""Load attention model weights."""
8888
path = str(axl_filename(pathlib.Path(directory) / filename))
89-
weights = torch.load(path, map_location=torch.device("cpu"))
89+
weights = torch.load(
90+
path, map_location=torch.device("cpu"), weights_only=True
91+
)
9092
return weights

axelrod/tests/strategies/test_attention.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,3 @@ def test_versus_alternator(self):
106106
def test_versus_handshake(self):
107107
actions = [(C, C), (C, D), (C, D), (D, D), (D, D), (C, D)]
108108
self.versus_test(axl.Handshake(), expected_actions=actions)
109-
110-
def test_versus_hopeless(self):
111-
actions = [(C, D), (C, C), (D, D), (C, C), (C, D), (D, C)]
112-
self.versus_test(axl.Hopeless(), expected_actions=actions)

0 commit comments

Comments
 (0)