diff --git a/libs/langchain/langchain/utilities/spark_sql.py b/libs/langchain/langchain/utilities/spark_sql.py index ffecbe511d4..20c1e8e5b2f 100644 --- a/libs/langchain/langchain/utilities/spark_sql.py +++ b/libs/langchain/langchain/utilities/spark_sql.py @@ -179,14 +179,8 @@ class SparkSQL: If the statement throws an error, the error message is returned. """ - try: - from pyspark.errors import PySparkException - except ImportError: - raise ValueError( - "pyspark is not installed. Please install it with `pip install pyspark`" - ) try: return self.run(command, fetch) - except PySparkException as e: + except Exception as e: """Format the error message""" return f"Error: {e}"