refactor: remove unnecessary code #5

twndus · May 9, 2024 · 00e97a5 · 00e97a5
1 parent e09247a
commit 00e97a5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/data/datasets/cdae_data_pipeline.py b/data/datasets/cdae_data_pipeline.py
@@ -18,10 +18,9 @@ def split(self, df):
         '''
         logger.info(f'start random user split...')
 
-        item_list = df.columns[1:].astype(int)
         train_data, valid_data, test_data = {}, {}, {}
 
-        for i, row in tqdm(df.iterrows(), total=df.shape[0]):
+        for _, row in tqdm(df.iterrows(), total=df.shape[0]):
 
             # train+valid+test mask
             user_id = int(row['user_id'])