refactor(nbia): use headers property instead of api_header attribute

jjjermiah · Jan 27, 2024 · c3b0940 · c3b0940
1 parent fe9a82b
commit c3b0940
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/nbiatoolkit/nbia.py b/src/nbiatoolkit/nbia.py
@@ -42,7 +42,7 @@ def __init__(self,
             "Setting up OAuth2 client... with username %s", username)
 
         self._oauth2_client = OAuth2(username=username, password=password)
-        self.api_headers = self._oauth2_client.getToken()
+        self._api_headers = self._oauth2_client.getToken()
 
     @property
     def headers(self):
@@ -56,7 +56,7 @@ def query_api(self, endpoint: NBIA_ENDPOINTS, params: dict = {}) -> dict:
         try:
             response = requests.get(
                 url=query_url,
-                headers=self.api_headers,
+                headers=self.headers,
                 params=params
                 )
             if response.headers.get('Content-Type') == 'application/json':