From 66cbbfa6d731b9496a238d7934ba592e0fb47357 Mon Sep 17 00:00:00 2001 From: Marc Harper Date: Tue, 10 Mar 2020 21:18:25 -0700 Subject: [PATCH] Suppress expected numpy warnings --- axelrod/result_set.py | 25 +++++++++++++++---------- 1 file changed, 15 insertions(+), 10 deletions(-) diff --git a/axelrod/result_set.py b/axelrod/result_set.py index 36505dd4d..a43a14af5 100644 --- a/axelrod/result_set.py +++ b/axelrod/result_set.py @@ -2,6 +2,7 @@ import csv import itertools from multiprocessing import cpu_count +import warnings import numpy as np import tqdm @@ -405,12 +406,14 @@ def _build_initial_cooperation_count(self, initial_cooperation_count_series): @update_progress_bar def _build_normalised_cooperation(self): - normalised_cooperation = [ - list(np.nan_to_num(row)) - for row in np.array(self.cooperation) - / sum(map(np.array, self.match_lengths)) - ] - return normalised_cooperation + with warnings.catch_warnings(): + warnings.simplefilter("ignore") + normalised_cooperation = [ + list(np.nan_to_num(row)) + for row in np.array(self.cooperation) + / sum(map(np.array, self.match_lengths)) + ] + return normalised_cooperation @update_progress_bar def _build_initial_cooperation_rate(self, interactions_series): @@ -420,10 +423,12 @@ def _build_initial_cooperation_rate(self, interactions_series): for player_index in range(self.num_players) ] ) - initial_cooperation_rate = list( - np.nan_to_num(np.array(self.initial_cooperation_count) / interactions_array) - ) - return initial_cooperation_rate + with warnings.catch_warnings(): + warnings.simplefilter("ignore") + initial_cooperation_rate = list( + np.nan_to_num(np.array(self.initial_cooperation_count) / interactions_array) + ) + return initial_cooperation_rate @update_progress_bar def _build_ranking(self):