logging

equinor · Dec 11, 2024 · afd918b · afd918b
1 parent c8873ac
commit afd918b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/warmth/build.py b/warmth/build.py
@@ -735,15 +735,16 @@ def _create_nodes(self, all_sediments_grid: List[List[List]]):
         """
         indexer = self.grid.indexing_arr
 
-
+        valid = 0
+        dropped = 0
         for index in indexer:
-
             if self.nodes[index[0]][index[1]] != False:
                 node_sed: list[_sediment_layer_] = []
                 for sed_grid in all_sediments_grid:
                     node_sed.append(sed_grid[index[0]][index[1]])
                 if all(node_sed) is False:
                     self.nodes[index[0]][index[1]] = False
+                    dropped += 1
                     logger.warning(f"dropping node {index}. One of the layer has no depth value")
                 else:
                     top = np.empty(0)
@@ -774,6 +775,7 @@ def _create_nodes(self, all_sediments_grid: List[List[List]]):
 
                     if checker is False:
                         self.nodes[index[0]][index[1]] = False
+                        dropped += 1
                     else:
                         df = self._fix_nan_sed(df)
                         n = single_node()
@@ -782,8 +784,10 @@ def _create_nodes(self, all_sediments_grid: List[List[List]]):
                         n.sediments_inputs=df
                         n.indexer = index
                         self.nodes[index[0]][index[1]] = n
+                        valid += 1
             else:
                 pass
+        logger.info(f"Dropped {dropped} nodes. Remaining valid nodes = {valid}")
         return
 
     def _check_nan_sed(self,df:pd.DataFrame)-> bool: