diff --git a/hyperglass/models/data/__init__.py b/hyperglass/models/data/__init__.py index 02d8f27..d19d5e8 100644 --- a/hyperglass/models/data/__init__.py +++ b/hyperglass/models/data/__init__.py @@ -4,11 +4,12 @@ from typing import Union # Local -from .bgp_route import BGPRouteTable +from .bgp_route import BGPRoute, BGPRouteTable OutputDataModel = Union[BGPRouteTable] __all__ = ( + "BGPRoute", "BGPRouteTable", "OutputDataModel", )