From 611ee1761af059c0cecb1082a2c9cc4870e6a793 Mon Sep 17 00:00:00 2001
From: Dom <97384583+tosemml@users.noreply.github.com>
Date: Tue, 8 Aug 2023 18:48:29 -0700
Subject: [PATCH] use np.cumsum

---
 data_prep/baselines/GREAT/models/util.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/data_prep/baselines/GREAT/models/util.py b/data_prep/baselines/GREAT/models/util.py
index 107d8d0..7e521a5 100644
--- a/data_prep/baselines/GREAT/models/util.py
+++ b/data_prep/baselines/GREAT/models/util.py
@@ -9,6 +9,4 @@ def positional_encoding(dim, sentence_length, dtype=tf.float32):
 	return tf.constant(encoded_vec.reshape([sentence_length, dim]), dtype=dtype)
 
 def prefix_sum(arr):
-	res = [0]
-	for a in arr: res.append(res[-1] + a)
-	return res
+	return [0] + np.cumsum(arr).tolist()