Merge pull request #50 from astropy/algolia_6

AlgoliaIndex: alter return
astropy · Nov 1, 2024 · d3a3e94 · d3a3e94
2 parents 96cf7e5 + bdf9aaa
commit d3a3e94
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/astropylibrarian/algolia/client.py b/astropylibrarian/algolia/client.py
@@ -67,10 +67,10 @@ class AlgoliaIndex(BaseAlgoliaIndex):
         Name of the Algolia index.
     """
 
-    async def __aenter__(self) -> SearchClient:
+    async def __aenter__(self) -> "AlgoliaIndex":
         self._logger.debug("Opening algolia client")
         self.algolia_client = SearchClient(self.app_id, self._key)
-        return self.algolia_client
+        return self
 
     async def __aexit__(
         self,