From b2450fa8be202941bb1b3770267fe6b9215d4c27 Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Thu, 3 Sep 2026 11:20:32 -0700
Subject: pytest tests -vs RUNS, but obv doesn't all PASS

---
 tests/incite/collections/test_df_collection_thl_marketplaces.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'tests/incite/collections/test_df_collection_thl_marketplaces.py')

diff --git a/tests/incite/collections/test_df_collection_thl_marketplaces.py b/tests/incite/collections/test_df_collection_thl_marketplaces.py
index 6ad0cb4..0f79b81 100644
--- a/tests/incite/collections/test_df_collection_thl_marketplaces.py
+++ b/tests/incite/collections/test_df_collection_thl_marketplaces.py
@@ -1,3 +1,4 @@
+from collections.abc import Generator
 from datetime import UTC, datetime
 from itertools import product
 from typing import TYPE_CHECKING
@@ -18,8 +19,8 @@ if TYPE_CHECKING:
     from generalresearch.pg_helper import PostgresConfig
 
 
-def combo_object():
-    for x in product(
+def combo_object() -> Generator[tuple[type, str]]:
+    yield from product(
         [
             InnovateSurveyHistoryCollection,
             MorningSurveyTimeseriesCollection,
@@ -27,8 +28,7 @@ def combo_object():
             SpectrumSurveyTimeseriesCollection,
         ],
         ["5min", "6H", "30D"],
-    ):
-        yield from x
+    )
 
 
 @pytest.mark.parametrize("df_coll, offset", combo_object())
-- 
cgit v1.2.3