You can probably ignore this comment, but can you move this import to be the first of this block? This way, we have all the local imports as the last ones, which I think is what PEP8 requires.