diff --git a/multidms/data.py b/multidms/data.py index 14d767b..14a7313 100644 --- a/multidms/data.py +++ b/multidms/data.py @@ -481,7 +481,6 @@ def get_nis_from_site_map(site_map): if "weight" in condition_func_score_df.columns: w[condition] = jnp.array(condition_func_score_df["weight"].values) - df.drop(["wts", "sites", "muts"], axis=1, inplace=True) self._variants_df = df self._training_data = {"X": X, "y": y, "w": w}