fix type checking

cleanlab · Aug 27, 2024 · ab54729 · ab54729
1 parent 9c1322f
commit ab54729
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cleanlab_studio/studio/trustworthy_language_model.py b/cleanlab_studio/studio/trustworthy_language_model.py
@@ -546,7 +546,7 @@ async def get_trustworthiness_score_async(
         processed_response = process_response_and_kwargs(response, kwargs)
 
         async with aiohttp.ClientSession() as session:
-            if isinstance(prompt, str) and isinstance(processed_response, str):
+            if isinstance(prompt, str) and isinstance(processed_response, dict):
                 trustworthiness_score = await self._get_trustworthiness_score_async(
                     prompt,
                     processed_response,