`ModuleNotFoundError` when using `policyengine` Python package in Colab

[MaxGhenis]

See https://colab.research.google.com/drive/1vBoDTq4sB7MlyM1rhHRoNaEgq5K4Fq4z?usp=sharing

!pip install policyengine

from policyengine.countries import US

ModuleNotFoundError Traceback (most recent call last) in () ----> 1 from policyengine.countries import US

1 frames /usr/local/lib/python3.7/dist-packages/policyengine/countries/country.py in () 8 ) 9 from openfisca_core.simulation_builder import SimulationBuilder ---> 10 from policyengine.impact.population.breakdown import ( 11 get_breakdown_and_chart_per_provision, 12 )

ModuleNotFoundError: No module named 'policyengine.impact'

[nikhilwoodruff]

Ah, looks like we need to add init.py files in some places still.