From afa285db08643203989b30a0a057280c979a36c6 Mon Sep 17 00:00:00 2001 From: Qiusheng Wu Date: Sun, 28 Apr 2024 23:58:15 -0400 Subject: [PATCH] Improve google_maps_api_key function --- geemap/common.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/geemap/common.py b/geemap/common.py index f147dfc533..a0c6f8ab5c 100644 --- a/geemap/common.py +++ b/geemap/common.py @@ -16325,12 +16325,12 @@ def google_map_tiles( return gmap_providers.get(map_type) -def google_maps_api_key(token_name="MAPS_API_KEY"): +def google_maps_api_key(lookup_key: str = "MAPS_API_KEY") -> Optional[str]: """ Retrieves the Google Maps API key from the environment or Colab user data. Args: - token_name (str, optional): The name of the environment variable or + lookup_key (str, optional): The name of the environment variable or Colab user data key where the API key is stored. Defaults to 'MAPS_API_KEY'. @@ -16340,8 +16340,8 @@ def google_maps_api_key(token_name="MAPS_API_KEY"): if in_colab_shell(): from google.colab import userdata - MAPS_API_KEY = userdata.get(token_name) + env_value = userdata.get(lookup_key) else: - MAPS_API_KEY = os.environ.get(token_name, None) + env_value = os.environ.get(lookup_key, None) - return MAPS_API_KEY + return env_value