almost there

visuals/components.py

@@ -4,7 +4,7 @@ import pandas as pd
 import logging
 import sys
 import config.exit_codes as ec
-import datetime
+# import datetime
 
 try:
     accounts = pl.read_parquet('data/warehouse/accounts.parquet')
@@ -30,9 +30,7 @@ except Exception as e:
 
 def update_dates(start_date, end_date):
     master_data = master_transactions.filter(
-        (pl.col('year') >= start_date.year ) & (pl.col('year') <= end_date.year) & 
-        (pl.col('month') >= start_date.month ) & (pl.col('month') <= end_date.month) &
-        (pl.col('day') >= start_date.day ) & (pl.col('day') <= end_date.day)
+        pl.col("date").is_between(start_date, end_date)
     )
     return master_data
 
@@ -119,5 +117,5 @@ def update_data(master_data):
             "spend_per_category_bar": spend_per_category_bar,
             "spend_per_payee_bar": spend_per_payee_bar,
             "total_spend": total_spend}
-    print(data)
+
     return data