[TLM] Fix get_trustworthiness_score_async type check (#299)

cleanlab · Aug 27, 2024 · 5ed28d0 · 5ed28d0
1 parent 9c1322f
commit 5ed28d0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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,

diff --git a/cleanlab_studio/version.py b/cleanlab_studio/version.py
@@ -1,7 +1,7 @@
 # Note to developers:
 # Consider if backend's MIN_CLI_VERSION needs updating when pushing any changes to this file.
 
-__version__ = "2.4.0"
+__version__ = "2.4.1"
 
 SCHEMA_VERSION = "0.2.0"
 MIN_SCHEMA_VERSION = "0.1.0"