From 2c74676902da44ff1f9262204200dd3201092978 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Thu, 15 Sep 2022 17:31:50 +0100
Subject: [PATCH] =?UTF-8?q?predict=20=E2=86=92=20predict=5Fon=5Fbatch?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 aimodel/src/lib/ai/RainfallWaterContraster.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aimodel/src/lib/ai/RainfallWaterContraster.py b/aimodel/src/lib/ai/RainfallWaterContraster.py
index 39b2db8..fbde335 100644
--- a/aimodel/src/lib/ai/RainfallWaterContraster.py
+++ b/aimodel/src/lib/ai/RainfallWaterContraster.py
@@ -88,7 +88,7 @@ class RainfallWaterContraster(object):
 		i_batch = -1
 		for batch in dataset:
 			i_batch += 1
-			rainfall = self.model_predict.predict(batch) # ((rainfall, water), dummy_label)
+			rainfall = self.model_predict.predict_on_batch(batch) # ((rainfall, water), dummy_label)
 			
 			for step in tf.unstack(rainfall, axis=0):
 				yield step