update sql_endpoint to work with the new pipeline
This commit is contained in:
@@ -134,6 +134,11 @@ class SqlEndpoint():
|
|||||||
"""
|
"""
|
||||||
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)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__" :
|
if __name__ == "__main__" :
|
||||||
|
|||||||
Reference in New Issue
Block a user