nesticot commited on
Commit
515500e
·
verified ·
1 Parent(s): 7b7ed65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ level = 'mlb'
94
  # dataset = load_dataset('TJStatsApps/mlb_data', data_files=[f'{level}_pitch_data_{season}.csv' ])
95
  dataset = pd.read_parquet(f"hf://datasets/TJStatsApps/mlb_data/data/{level}_pitch_data_{season}.parquet")
96
  # dataset_train = dataset['train']
97
- df = dataset.set_index(list(dataset_train.features.keys())[0]).reset_index(drop=True).drop_duplicates(subset=['play_id'],keep='last')
98
  df['batter_name_team'] = df['batter_name'] + ' - ' + df['batter_team']
99
 
100
  # Sample dictionary
 
94
  # dataset = load_dataset('TJStatsApps/mlb_data', data_files=[f'{level}_pitch_data_{season}.csv' ])
95
  dataset = pd.read_parquet(f"hf://datasets/TJStatsApps/mlb_data/data/{level}_pitch_data_{season}.parquet")
96
  # dataset_train = dataset['train']
97
+ df = dataset.drop_duplicates(subset=['play_id'],keep='last')
98
  df['batter_name_team'] = df['batter_name'] + ' - ' + df['batter_team']
99
 
100
  # Sample dictionary