fixes from restack

MSeal · Nov 14, 2023 · 8bc0e6f · 8bc0e6f
1 parent 61e843a
commit 8bc0e6f
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/src/dx/formatters/main.py b/src/dx/formatters/main.py
@@ -23,13 +23,6 @@
 )
 from dx.utils.tracking import DXDF_CACHE, SUBSET_HASH_TO_PARENT_DATA, DXDataFrame, get_db_connection
 
-try:
-    from repr_llm.pandas import summarize_dataframe
-
-    REPR_LLM_INSTALLED = True
-except ImportError:
-    REPR_LLM_INSTALLED = False
-
 logger = structlog.get_logger(__name__)
 db_connection = get_db_connection()
 settings = get_settings()
@@ -217,7 +210,7 @@ def format_output(
     try:
         payload["text/llm+plain"] = make_df_summary(df)
     except Exception as e:
-        logger.debug(f"Error in summarize_dataframe: {e}")
+        logger.debug(f"Error making dataframe summary: {e}")
 
     metadata = {settings.MEDIA_TYPE: metadata}