Import python implementations from python, not rust wrapper

stac-utils · Oct 18, 2024 · 37eaee6 · 37eaee6
1 parent 4827cee
commit 37eaee6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git 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
@@ -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(