From 1da2ae579f3fb85de12f8ddc4d8c4c41826edbb8 Mon Sep 17 00:00:00 2001 From: Diogo Sousa Date: Sat, 29 Jul 2023 04:15:57 +0100 Subject: [PATCH] private methods --- .../sdk/provider/openbb_provider/registry_map.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/openbb_sdk/sdk/provider/openbb_provider/registry_map.py b/openbb_sdk/sdk/provider/openbb_provider/registry_map.py index 94ec3fdf943..e0a7830a5a0 100644 --- a/openbb_sdk/sdk/provider/openbb_provider/registry_map.py +++ b/openbb_sdk/sdk/provider/openbb_provider/registry_map.py @@ -11,10 +11,10 @@ class RegistryMap: def __init__(self, registry: Optional[Registry] = None) -> None: self._registry = registry or RegistryLoader.from_extensions() - self._required_credentials = self.get_required_credentials(self._registry) - self._available_providers = self.get_available_providers(self._registry) - self._map = self.get_map(self._registry) - self._models = self.get_models(self._map) + self._required_credentials = self._get_required_credentials(self._registry) + self._available_providers = self._get_available_providers(self._registry) + self._map = self._get_map(self._registry) + self._models = self._get_models(self._map) @property def registry(self) -> Registry: @@ -36,7 +36,7 @@ class RegistryMap: def models(self) -> List[str]: return self._models - def get_required_credentials(self, registry: Registry) -> List[str]: + def _get_required_credentials(self, registry: Registry) -> List[str]: """Get list of required credentials.""" cred_list = [] for provider in registry.providers.values(): @@ -44,11 +44,11 @@ class RegistryMap: cred_list.append(c) return cred_list - def get_available_providers(self, registry: Registry) -> List[str]: + def _get_available_providers(self, registry: Registry) -> List[str]: """Get list of available providers.""" return sorted(list(registry.providers.keys())) - def get_map(self, registry: Registry) -> MapType: + def _get_map(self, registry: Registry) -> MapType: """Generate map for the provider package.""" map_: MapType = {} @@ -72,7 +72,7 @@ class RegistryMap: return map_ - def get_models(self, map_: MapType) -> List[str]: + def _get_models(self, map_: MapType) -> List[str]: """Get available models.""" return list(map_.keys())