prevent div 0

BioSTEAMDevelopmentGroup · Jun 27, 2024 · f80cec3 · f80cec3
1 parent b6e4e29
commit f80cec3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/thermosteam/_stream.py b/thermosteam/_stream.py
@@ -2659,7 +2659,10 @@ def _info_df(self, units, notation, composition, N_max, all_IDs, indexer, factor
                 total_flow = flow_array.sum()
                 index.append((f"{phase} [{flow_units}]", ''))
                 data.append(f"{total_flow:{flow_notation}}")
-                comp_array = 100 * flow_array / total_flow
+                if total_flow == 0:
+                    comp_array = flow_array
+                else:
+                    comp_array = 100 * flow_array / total_flow
                 for i, (ID, comp) in enumerate(zip(all_IDs, comp_array)):
                     if not comp: continue
                     if i >= N_max: