Use six for input compat, support for dirs with both .py and .ipynb files.