remove positional argument in df.drop() (compat with pandas 2.x)

jan-spr · jan-spr · commit eac627d54733 · 2026-01-09T16:07:48.000+01:00
diff --git a/mofapy2/core/BayesNet.py b/mofapy2/core/BayesNet.py
@@ -770,7 +770,7 @@ def iterate(self):
             "time": iter_time,
             "number_factors": number_factors,
             "elbo": elbo["total"].values,
-            "elbo_terms": elbo.drop("total", 1),
+            "elbo_terms": elbo.drop("total", axis=1),
         }
         if "Sigma" in self.nodes.keys():
             tmp = self.nodes["Sigma"].getParameters()  # save only last iteration

Original file line number	Diff line number	Diff line change
`@@ -770,7 +770,7 @@ def iterate(self):`
`770`	`770`	`"time": iter_time,`
`771`	`771`	`"number_factors": number_factors,`
`772`	`772`	`"elbo": elbo["total"].values,`
`773`		`- "elbo_terms": elbo.drop("total", 1),`
	`773`	`+ "elbo_terms": elbo.drop("total", axis=1),`
`774`	`774`	`}`
`775`	`775`	`if "Sigma" in self.nodes.keys():`
`776`	`776`	`tmp = self.nodes["Sigma"].getParameters() # save only last iteration`