typo

Hitchwiki · Jan 12, 2025 · 41a5c66 · 41a5c66
1 parent bac7651
commit 41a5c66
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/heatchmap/gpmap.py b/heatchmap/gpmap.py
@@ -153,16 +153,16 @@ def recalc_map(self):
                 if len(to_predict) == self.batch_size:
                     waiting_times, uncertainties = self.gpr.predict(np.array(to_predict), return_std=True)
                     for i, (y, x) in enumerate(pixels_to_predict):
-                        self.raw_raster[y][x] = waiting_time[i]
+                        self.raw_raster[y][x] = waiting_times[i]
                         self.uncertainties[y][x] = uncertainties[i]
 
                     to_predict = []
                     pixels_to_predict = []
 
         if len(to_predict) > 0:
-            waiting_time, uncertainties = self.gpr.predict(np.array(to_predict), return_std=True)
+            waiting_times, uncertainties = self.gpr.predict(np.array(to_predict), return_std=True)
             for i, (y, x) in enumerate(pixels_to_predict):
-                self.raw_raster[y][x] = waiting_time[i]
+                self.raw_raster[y][x] = waiting_times[i]
                 self.uncertainties[y][x] = uncertainties[i]
 
 

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 import setuptools
 
-VERSION = "0.1.26"
+VERSION = "0.1.27"
 
 NAME = "heatchmap"