fix the dataframe code

HopkinsIDD · Dec 3, 2024 · 956028d · 956028d
1 parent 8ea5782
commit 956028d
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/flepimop/gempyor_pkg/src/gempyor/inference_parameter.py b/flepimop/gempyor_pkg/src/gempyor/inference_parameter.py
@@ -175,17 +175,13 @@ def __len__(self):
         return len(self.pnames)
 
     def get_parameter_df(self):
-        all_params = []
-
-        for i in zip(self.ptypes, self.subpops,self.pnames):
-            data = {
-                "ptypes": self.ptypes,
-                "subpops": self.subpops,
-                "pnames": self.pnames
-            }
-            all_params.append(pd.DataFrame(data))
-
-        return pd.concat(all_params)
+        data = {
+            "ptypes": self.ptypes,
+            "subpops": self.subpops,
+            "pnames": self.pnames
+        }
+        df = pd.DataFrame(data)
+        return df
 
     def draw_initial(self, n_draw=1):
         """