ngs data pull updated

handles specified ngs seasons correctly now
nflverse · Jul 30, 2022 · bcff032 · bcff032
1 parent 7e6e342
commit bcff032
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/nfl_data_py/__init__.py b/nfl_data_py/__init__.py
@@ -658,12 +658,11 @@ def import_ngs_data(stat_type, years=None):
         raise ValueError('years variable must be list or range.')
 
     # import data
-    if len(years) == 0:
-        url = r'https://github.com/nflverse/nflverse-data/releases/download/nextgen_stats/ngs_{0}.parquet'.format(stat_type)
-        data = pandas.read_parquet(url)
-    else:
-        url = r'https://github.com/nflverse/nflverse-data/releases/download/nextgen_stats/ngs_{0}_{1}.parquet'
-        data = pandas.concat([pandas.read_parquet(url.format(x, stat_type), engine='auto') for x in years])
+    url = r'https://github.com/nflverse/nflverse-data/releases/download/nextgen_stats/ngs_{0}.parquet'.format(stat_type)
+    data = pandas.read_parquet(url)
+
+    if len(years) > 0:
+        data = data[data['season'].isin([x for x in years])]
 
     # return
     return data

diff --git a/nfl_data_py/__pycache__/__init__.cpython-39.pyc b/nfl_data_py/__pycache__/__init__.cpython-39.pyc