diff --git a/python/python/stacrs/__init__.py b/python/python/stacrs/__init__.py index 1abac7c4..d65c5839 100644 --- a/python/python/stacrs/__init__.py +++ b/python/python/stacrs/__init__.py @@ -1,16 +1,19 @@ -from . import stacrs -from .stacrs import ( +import pystac +from .pystac import ( migrate, - migrate_href, read, search, - search_to, validate, - validate_href, write, ) -__doc__ = stacrs.__doc__ +from .stacrs import ( + migrate_href, + search_to, + validate_href, +) + +__doc__ = pystac.__doc__ __all__ = [ "migrate", "migrate_href", diff --git a/python/python/stacrs/pystac.py b/python/python/stacrs/pystac.py index ee14aae2..ac756253 100644 --- a/python/python/stacrs/pystac.py +++ b/python/python/stacrs/pystac.py @@ -149,7 +149,7 @@ def search( ... max_items=1, ... )) """ - items = stacrs.search( + items_dict = stacrs.search( href, intersects=intersects, ids=ids, @@ -164,7 +164,7 @@ def search( filter=filter, query=query, ) - return (Item.from_dict(d) for d in items) + return ItemCollection(Item.from_dict(d) for d in items_dict) def validate(