From cb4f3c255e3caf47fcca9424b7dc4d8b47ed574a Mon Sep 17 00:00:00 2001
From: Giorgio Caizzi <64302135+giocaizzi@users.noreply.github.com>
Date: Sat, 22 Jul 2023 02:51:05 +0200
Subject: [PATCH] quickfix normalized grid in modeler (#40)

---
 py3dinterpolations/__init__.py          | 2 +-
 py3dinterpolations/modelling/modeler.py | 4 ++--
 setup.py                                | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/py3dinterpolations/__init__.py b/py3dinterpolations/__init__.py
index dd28fe8..acc2b00 100644
--- a/py3dinterpolations/__init__.py
+++ b/py3dinterpolations/__init__.py
@@ -1,7 +1,7 @@
 """quick 3D interpolation with python"""
 
 # Version
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 # Credits
 __author__ = "Giorgio Caizzi"
diff --git a/py3dinterpolations/modelling/modeler.py b/py3dinterpolations/modelling/modeler.py
index 86501c4..045eb57 100644
--- a/py3dinterpolations/modelling/modeler.py
+++ b/py3dinterpolations/modelling/modeler.py
@@ -78,9 +78,9 @@ def predict(self, **kwargs) -> np.ndarray:
         """
         # make predictions on normalized grid if normalization was applied
         if "normalization" in self.griddata.preprocessor_params.keys():
-            grids_arrays = self.grid3d.grid
-        else:
             grids_arrays = self.grid3d.normalized_grid
+        else:
+            grids_arrays = self.grid3d.grid
 
         # predict
         interpolated, variance = self.model.predict(
diff --git a/setup.py b/setup.py
index 13e2c1d..dddd4a9 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,7 @@
 
 setup(
     name="py3dinterpolations",
-    version="0.3.0",
+    version="0.3.1",
     description="quick 3D interpolation with python",
     long_description_content_type="text/markdown",
     long_description=long_description,