from .basic import Rectangle, Square, Ellipse, Circle  # noqa: F401
from .shape import Shape  # noqa: F401

from .basic import Rectangle, Square, Ellipse, Circle, Triangle  # noqa: F401
from .shape import Shape  # noqa: F401

Add Triangle class to __init__.py