diff --git a/dataloader.py b/dataloader.py index edbb7f8e89c24ca49707ff325493bc9dc870ba53..2ccd88c69e44b9f811a64eddabc94cf994f7a0e4 100644 --- a/dataloader.py +++ b/dataloader.py @@ -255,6 +255,7 @@ class dataloader: self.df = pd.merge(self.df, r_df, how="left") else: self.df = r_df + print(f" data size is {self.df.shape}") # define ids self.ids = list(self.df.id.unique()) if not use_tsv: