clean up

Author: jonhealy1

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/core.py

@@ -72,9 +72,7 @@ def all_collections(self, **kwargs) -> Collections:
     def get_collection(self, collection_id: str, **kwargs) -> Collection:
         """Get collection by id."""
         base_url = str(kwargs["request"].base_url)
-        collection = self.database.find_collection(collection_id=collection_id)[
-            "_source"
-        ]
+        collection = self.database.find_collection(collection_id=collection_id)
         return self.collection_serializer.db_to_stac(collection, base_url)
 
     @overrides

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/database_logic.py

@@ -283,7 +283,7 @@ def find_collection(self, collection_id: str) -> Collection:
         except elasticsearch.exceptions.NotFoundError:
             raise NotFoundError(f"Collection {collection_id} not found")
 
-        return collection
+        return collection["_source"]
 
     def delete_collection(self, collection_id: str):
         """Database logic for deleting one collection."""