This should be before the numpy imports with a blank line between them.