fix: update constant

Mause · Dec 5, 2023 · ec61da2 · ec61da2
1 parent edeeae8
commit ec61da2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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)