From ec61da22ae6251814c604e6dc04cfa5c6de40444 Mon Sep 17 00:00:00 2001 From: Elliana May Date: Tue, 5 Dec 2023 09:57:55 +0800 Subject: [PATCH] fix: update constant --- duckdb_engine/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/duckdb_engine/__init__.py b/duckdb_engine/__init__.py index 1f3b45e0..3516bdb4 100644 --- a/duckdb_engine/__init__.py +++ b/duckdb_engine/__init__.py @@ -136,7 +136,11 @@ def execute( try: if statement.lower() == "commit": # this is largely for ipython-sql self.__c.commit() - elif statement.lower() in ("register", "register(?, ?)"): + elif statement.lower() in ( + "register", + "register(?, ?)", + "register($1, $2)", + ): assert parameters and len(parameters) == 2, parameters view_name, df = parameters self.__c.register(view_name, df)