lint and clean up

tests/scorers/test_llm_integrations.py

@@ -1,4 +1,5 @@
 import os
+
 import pytest
 
 from weave.scorers.summarization_scorer import (
@@ -26,13 +27,14 @@
     },
 }
 
+
 @pytest.fixture(
     params=[
         (provider, model)
         for provider, cfg in PROVIDERS.items()
         for model in cfg["models"]
     ],
-    ids=lambda p: f"{p[0]}:{p[1]}"
+    ids=lambda p: f"{p[0]}:{p[1]}",
 )
 def summarization_scorer(request):
     """
@@ -49,6 +51,7 @@ def summarization_scorer(request):
         max_tokens=1024,
     )
 
+
 @pytest.mark.asyncio
 async def test_summarization_scorer_evaluate_summary(summarization_scorer):
     input_text = "The wolf is lonely in the forest. He is not happy that the fox is not with him."

weave/scorers/initialization.py

@@ -5,9 +5,9 @@ def check_litellm_installation():
     """Check if litellm is installed and raise an informative error if not."""
     try:
         from litellm import acompletion  # noqa: F401
-
-        return acompletion
     except ImportError:
         raise ImportError(
             "litellm is required to use the LLM-powered scorers, please install it with `pip install litellm`"
         )
+    else:
+        return True