from marshmallow_polyfield.polyfield import PolyField, PolyFieldBase  # noqa

from marshmallow_polyfield.polyfield import PolyField, PolyFieldBase

__all__ = ['PolyField', 'PolyFieldBase']

add __all__ to remove noqa comment