fix: unit tests

Author: Darinochka

tests/callback/test_callback.py

@@ -146,6 +146,7 @@ def test_pipeline_callbacks(dataset):
                         "query_prompt": None,
                         "sts_prompt": None,
                         "use_cache": False,
+                        "similarity_fn_name": "cosine",
                     },
                     "k": 1,
                     "weights": "uniform",
@@ -180,6 +181,7 @@ def test_pipeline_callbacks(dataset):
                         "query_prompt": None,
                         "sts_prompt": None,
                         "use_cache": False,
+                        "similarity_fn_name": "cosine",
                     },
                     "k": 1,
                     "weights": "distance",
@@ -214,6 +216,7 @@ def test_pipeline_callbacks(dataset):
                         "query_prompt": None,
                         "sts_prompt": None,
                         "use_cache": False,
+                        "similarity_fn_name": "cosine",
                     },
                 },
                 "module_name": "linear",