diff --git a/tests/components/business_logic/test_pinning_business_logic.py b/tests/components/business_logic/test_pinning_business_logic.py index f0b945f..665fc91 100644 --- a/tests/components/business_logic/test_pinning_business_logic.py +++ b/tests/components/business_logic/test_pinning_business_logic.py @@ -66,6 +66,7 @@ def __init__(self): entity_store={}, events=[], feature_map=FeatureMap(feature_map={}), + model_score_map={}, ), ) return await pipeline.execute(request) diff --git a/tests/scenarios/test_product_ranking.py b/tests/scenarios/test_product_ranking.py index 096df65..8cf4296 100644 --- a/tests/scenarios/test_product_ranking.py +++ b/tests/scenarios/test_product_ranking.py @@ -384,6 +384,7 @@ async def test_hydrate(mock_redis): json=json_input, headers={}, entity_store={}, + model_score_map={}, events=[], feature_map=FeatureMap(feature_map={}), ) @@ -447,6 +448,7 @@ async def test_hydrate__duplicate_brand(mock_redis__duplicate_brand): entity_store={}, events=[], feature_map=FeatureMap(feature_map={}), + model_score_map={}, ) request_context.set(test_wyvern_request)