update sql_endpoint to work with the new pipeline

This commit is contained in:
GassiGiuseppe 2025-10-05 14:58:03 +02:00
parent acb43fc899
commit 7307916891

View File

@ -133,6 +133,11 @@ class SqlEndpoint():
GROUP BY RelationshipURI; GROUP BY RelationshipURI;
""" """
return pd.read_sql_query(QUERY, self.sql_engine) return pd.read_sql_query(QUERY, self.sql_engine)
def get_dataframe_from_query(self, query: str, params=None):
if params is None:
return pd.read_sql_query(query, self.sql_engine)
return pd.read_sql_query(query, self.sql_engine, params=params)