From 20d466565ca75cae2102c42097eb821a1246d965 Mon Sep 17 00:00:00 2001 From: unknown Date: Mon, 27 Jan 2025 16:28:58 +0000 Subject: [PATCH] Fixed unnecessary column creation during variable init. --- src/hivpy/population.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/hivpy/population.py b/src/hivpy/population.py index fc45c7b..5e58750 100644 --- a/src/hivpy/population.py +++ b/src/hivpy/population.py @@ -121,11 +121,12 @@ def init_variable(self, name: str, init_val, n_prev_steps=0, data_type=None): to be stored (default 0). """ self.variable_history[name] = n_prev_steps + 1 - if data_type is not None: - self.data[name] = pd.Series([init_val]*self.size, dtype=data_type) + if (n_prev_steps == 0): + if data_type is not None: + self.data[name] = pd.Series([init_val]*self.size, dtype=data_type) + else: + self.data[name] = init_val else: - self.data[name] = init_val - if (n_prev_steps > 0): for i in range(0, n_prev_steps + 1): if data_type is not None: self.data[self.constructParamColumn(name, i)] = pd.Series([init_val]*self.size, dtype=data_type)