Export ._annotations.custom_attrs.

scrapy-plugins · Sep 19, 2024 · b7e22e7 · b7e22e7
1 parent ceb67a7
commit b7e22e7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/scrapy_zyte_api/__init__.py b/scrapy_zyte_api/__init__.py
@@ -5,7 +5,7 @@
 
     install_reactor("twisted.internet.asyncioreactor.AsyncioSelectorReactor")
 
-from ._annotations import ExtractFrom, actions
+from ._annotations import ExtractFrom, actions, custom_attrs
 from ._middlewares import (
     ScrapyZyteAPIDownloaderMiddleware,
     ScrapyZyteAPISpiderMiddleware,

diff --git a/tests/test_providers.py b/tests/test_providers.py
@@ -3,8 +3,6 @@
 
 import pytest
 
-from scrapy_zyte_api._annotations import custom_attrs
-
 pytest.importorskip("scrapy_poet")
 
 import attrs
@@ -36,7 +34,14 @@
 )
 from zyte_common_items.fields import auto_field
 
-from scrapy_zyte_api import Actions, ExtractFrom, Geolocation, Screenshot, actions
+from scrapy_zyte_api import (
+    Actions,
+    ExtractFrom,
+    Geolocation,
+    Screenshot,
+    actions,
+    custom_attrs,
+)
 from scrapy_zyte_api.handler import ScrapyZyteAPIDownloadHandler
 from scrapy_zyte_api.providers import _AUTO_PAGES, _ITEM_KEYWORDS, ZyteApiProvider