From dc0873cd2b004669f7f9496cd4ffb75a042251b7 Mon Sep 17 00:00:00 2001 From: Andy Eschbacher Date: Wed, 23 Mar 2016 15:33:54 -0400 Subject: [PATCH] add types to arrays --- src/py/crankshaft/crankshaft/space_time_dynamics/markov.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/py/crankshaft/crankshaft/space_time_dynamics/markov.py b/src/py/crankshaft/crankshaft/space_time_dynamics/markov.py index db2be6f..e809632 100644 --- a/src/py/crankshaft/crankshaft/space_time_dynamics/markov.py +++ b/src/py/crankshaft/crankshaft/space_time_dynamics/markov.py @@ -140,9 +140,9 @@ def get_prob_stats(prob_dist, unit_indices): """ num_elements = len(prob_dist) - trend_up = np.empty(num_elements) - trend_down = np.empty(num_elements) - trend = np.empty(num_elements) + trend_up = np.empty(num_elements, dtype=float) + trend_down = np.empty(num_elements, dtype=float) + trend = np.empty(num_elements, dtype=float) for i in range(num_elements): trend_up[i] = prob_dist[i, (unit_indices[i]+1):].sum()