diff --git a/nfl_data_py/__init__.py b/nfl_data_py/__init__.py index 2b3947a..307654a 100644 --- 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 index f67565e..2a79db8 100644 Binary files a/nfl_data_py/__pycache__/__init__.cpython-39.pyc and b/nfl_data_py/__pycache__/__init__.cpython-39.pyc differ