Skip to content

Commit

Permalink
fix for obbject extensions not added in extension map
Browse files Browse the repository at this point in the history
  • Loading branch information
the-praxs committed Mar 12, 2024
1 parent c13ac55 commit b6bd2b9
Showing 1 changed file with 7 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -137,10 +137,12 @@ def _get_extension_map(self) -> Dict[str, List[str]]:
groups = [
OpenBBGroups.core.value,
OpenBBGroups.provider.value,
OpenBBGroups.obbject.value,
]
entry_points_ = [
el.core_entry_points,
el.provider_entry_points,
el.obbject_entry_points,
]

for group, entry_point in zip(groups, entry_points_):
Expand Down Expand Up @@ -252,7 +254,11 @@ def _diff(path: Path) -> Tuple[Set[str], Set[str]]:

add: Set[str] = set()
remove: Set[str] = set()
groups = ("openbb_core_extension", "openbb_provider_extension")
groups = (
"openbb_core_extension",
"openbb_provider_extension",
"openbb_obbject_extension",
)
for g in groups:
built = set(ext_map.get(g, {}))
installed = set(
Expand Down Expand Up @@ -556,7 +562,6 @@ def format_params(
path: str, parameter_map: Dict[str, Parameter]
) -> OrderedDict[str, Parameter]:
"""Format the params."""

parameter_map.pop("cc", None)
# we need to add the chart parameter here bc of the docstring generation
if CHARTING_INSTALLED and path.replace("/", "_")[1:] in Charting.functions():
Expand Down

0 comments on commit b6bd2b9

Please sign in to comment.