data/__init__.py

import os
import polars as pl

from .seasons import SEASONS

assert (USE_CACHE := os.getenv('USE_CACHE', 'False')) in ('True', 'False')

if USE_CACHE == 'True':
  data_df = pl.read_parquet('data.parquet')
else:
  from .data import data_df