fix: filtered columns

rafoolin · Jan 21, 2024 · 41231d6 · 41231d6
1 parent aaf12bd
commit 41231d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/exercises/exercise4.py b/exercises/exercise4.py
@@ -105,7 +105,7 @@ def __extract(self, url: str, csv_file_name: str) -> None:
         if self.logging:
             print(f"CSV file path: {csv_file_path}")
         # Only read some of the columns
-        columns_to_read = [0, 1, 2, 3, 4, 7, 8, 9]
+        columns_to_read = [0, 1, 2, 3, 4, 8, 9]
         # Read the CSV file
         self.data_frame = pd.read_csv(
             csv_file_path,
@@ -214,7 +214,7 @@ def run_pipeline(self, url: str) -> None:
     ATTEMPTS = 1
     while ATTEMPTS < RETRY_THRESHOLD:
         try:
-            Pipeline(logging=True).run_pipeline(url=ZIP_URL)
+            Pipeline(logging=False).run_pipeline(url=ZIP_URL)
             break
         except Exception as e:
             ATTEMPTS += 1