Update spray.py

spray.py

@@ -18,11 +18,12 @@ from matplotlib.gridspec import GridSpec
 import matplotlib.pyplot as plt
 from scipy.stats import gaussian_kde
 
+import polars as pl
+
+season = 2025
+
 ### Import Datasets
-dataset = load_dataset('nesticot/mlb_data', data_files=['mlb_pitch_data_2023.csv',
-                                                        'mlb_pitch_data_2022.csv'])
-dataset_train = dataset['train']
-df_2023 = dataset_train.to_pandas().set_index(list(dataset_train.features.keys())[0]).reset_index(drop=True)
+df_2023 = pl.read_parquet(f"hf://datasets/TJStatsApps/mlb_data/data/mlb_pitch_data_2025.parquet").to_pandas()
 # Paths to data
 ### Normalize Hit Locations
 df_2023['hit_x'] = df_2023['hit_x'] - 126#df_2023['hit_x'].median()