From 858fbd89754cf7cd3d698a702253ac8c0d9f3111 Mon Sep 17 00:00:00 2001
From: Max <Hunt>
Date: Tue, 5 Jan 2021 15:25:41 +0000
Subject: [PATCH] Update

---
 dnup_virtual_sensor/virtual_soil_sensor.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dnup_virtual_sensor/virtual_soil_sensor.py b/dnup_virtual_sensor/virtual_soil_sensor.py
index 7aac165..e03686f 100755
--- a/dnup_virtual_sensor/virtual_soil_sensor.py
+++ b/dnup_virtual_sensor/virtual_soil_sensor.py
@@ -44,8 +44,8 @@ class VirtualProbe:
 
     def predict_soil_temp(self):
         feature = self.firebase.get_feature()
-        predicted_temp = self.model.predict([feature])
-        logging.debug(f'Predicted watering volume: {predicted_temp}')
+        predicted_temp = self.model.predict([feature])[0][0]
+        logging.debug(f'Predicted temp: {predicted_temp}')
         return predicted_temp