fix test

Skyvern-AI · Sep 11, 2023 · 6857098 · 6857098
1 parent 9153cb7
commit 6857098
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git 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
@@ -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)