```from statsmodels.regression.linear_model import OLS```

It doesn't make much difference in this case, but it's better to import from the actual module (inside statsmodels)