Could you consider adding py.typed to the module directory, and installing this file, so that mypy does not complain when importing opt_einsum?