James McCool commited on
Commit
c601cc3
·
1 Parent(s): 62ea802

Updating the database_queries for NASCAR to be for slate-specific.

Browse files
Files changed (1) hide show
  1. database_queries.py +2 -1
database_queries.py CHANGED
@@ -2057,7 +2057,7 @@ def init_nascar_baselines(type_var: str, site_var: str, slate_var: str):
2057
  if slate_var == 'Main':
2058
  slate_var = 'Main Slate'
2059
  elif slate_var == 'Secondary':
2060
- slate_var = 'Secondary Slate'
2061
  elif slate_var == 'Auxiliary':
2062
  slate_var = 'Late Slate'
2063
 
@@ -2098,6 +2098,7 @@ def init_nascar_baselines(type_var: str, site_var: str, slate_var: str):
2098
  cursor = collection.find()
2099
 
2100
  raw_display = pd.DataFrame(list(cursor))
 
2101
  raw_display = raw_display[['Player', 'Position', 'Grid', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id']]
2102
  raw_display['Team'] = 'NASCAR'
2103
  raw_display['Site'] = site_var
 
2057
  if slate_var == 'Main':
2058
  slate_var = 'Main Slate'
2059
  elif slate_var == 'Secondary':
2060
+ slate_var = 'Secondary'
2061
  elif slate_var == 'Auxiliary':
2062
  slate_var = 'Late Slate'
2063
 
 
2098
  cursor = collection.find()
2099
 
2100
  raw_display = pd.DataFrame(list(cursor))
2101
+ raw_display = raw_display[raw_display['slate'] == slate_var]
2102
  raw_display = raw_display[['Player', 'Position', 'Grid', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id']]
2103
  raw_display['Team'] = 'NASCAR'
2104
  raw_display['Site'] = site_var