From abf599d6ad29a71cf28243e5b6e69cea14380b62 Mon Sep 17 00:00:00 2001 From: Siarhei Siniak Date: Thu, 29 Jul 2021 22:18:54 +0300 Subject: [PATCH] [~] Refactor --- python/tasks/mlb_player.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py index 8cc2a02..340f2d4 100644 --- a/python/tasks/mlb_player.py +++ b/python/tasks/mlb_player.py @@ -1780,7 +1780,7 @@ def kernel_30(o_29, ids=None, delay=None,): pprint.pprint([t8.sum(), t8.mean()]) ids = numpy.random.choice( numpy.where(t8)[0], - min(70, len(t8)), + min(70, t8.sum()), ) elif ids == 'v3': t8 = numpy.stack( @@ -1794,7 +1794,7 @@ def kernel_30(o_29, ids=None, delay=None,): pprint.pprint([t8.sum(), t8.mean()]) ids = numpy.random.choice( numpy.where(t8)[0], - min(70, len(t8)), + min(70, t8.sum()), ) elif ids == 'v4': t8 = numpy.stack( @@ -1808,7 +1808,7 @@ def kernel_30(o_29, ids=None, delay=None,): pprint.pprint([t8.sum(), t8.mean()]) ids = numpy.random.choice( numpy.where(t8)[0], - min(70, len(t8)), + min(70, t8.sum()), ) else: assert isinstance(ids, numpy.ndarray)