Update app.py

app.py

@@ -6,6 +6,7 @@ import os
 import json
 import streamlit as st
 import pandas as pd
+import polars as pl
 from st_aggrid import AgGrid, GridOptionsBuilder, GridUpdateMode
 
 
@@ -92,8 +93,8 @@ season = 2025
 level = 'mlb'
 # dataset = load_dataset('TJStatsApps/mlb_data', data_files=[f'{level}_pitch_data_{season}.csv' ])
 dataset = pl.read_parquet(f"hf://datasets/TJStatsApps/mlb_data/data/{level}_pitch_data_{season}.parquet")
-dataset_train = dataset['train']
-df = dataset_train.to_pandas().set_index(list(dataset_train.features.keys())[0]).reset_index(drop=True).drop_duplicates(subset=['play_id'],keep='last')
+# dataset_train = dataset['train']
+df = dataset.to_pandas().set_index(list(dataset_train.features.keys())[0]).reset_index(drop=True).drop_duplicates(subset=['play_id'],keep='last')
 df['batter_name_team'] = df['batter_name'] + ' - ' + df['batter_team']
 
 # Sample dictionary