> from diagrams import Diagram

Slightly off-topic, but I wish JavaScript's import syntax was designed to be this way around, for better autocomplete in your editor.