diff --git a/.gitignore b/.gitignore
index 7e75fc8b..fc3631ed 100644
--- a/.gitignore
+++ b/.gitignore
@@ -171,3 +171,4 @@ Thumbs.db
 
 
 # End of https://www.toptal.com/developers/gitignore/api/python
+.vscode/*
\ No newline at end of file
diff --git a/src/python/tools/preprocess/datasets/ogbn_products.py b/src/python/tools/preprocess/datasets/ogbn_products.py
index bade27c6..d28e43ed 100644
--- a/src/python/tools/preprocess/datasets/ogbn_products.py
+++ b/src/python/tools/preprocess/datasets/ogbn_products.py
@@ -80,7 +80,6 @@ def preprocess(
             output_dir=self.output_directory,
             train_edges=self.input_edge_list_file,
             num_partitions=num_partitions,
-            columns=[0, 1],
             src_column=0,
             dst_column=1,
             remap_ids=remap_ids,