mplang-nightly 0.1.dev151__tar.gz → 0.1.dev152__tar.gz

{mplang_nightly-0.1.dev151 → mplang_nightly-0.1.dev152}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mplang-nightly
-Version: 0.1.dev151
+Version: 0.1.dev152
 Summary: Multi-Party Programming Language
 Author-email: SecretFlow Team <secretflow-contact@service.alipay.com>
 License:                                  Apache License

{mplang_nightly-0.1.dev151 → mplang_nightly-0.1.dev152}/mplang/kernels/sql_duckdb.py

@@ -25,6 +25,11 @@ def _duckdb_sql(pfunc: PFunction, *args: Any) -> Any:
     import duckdb
     import pandas as pd
 
+    # TODO: maybe we could translate the sql to duckdb dialect
+    # instead of raising an exception
+    if pfunc.attrs.get("dialect") != "duckdb":
+        raise ValueError("duckdb.run_sql must have dialect=duckdb attr")
+
     conn = duckdb.connect(":memory:")
     if args:
         in_names = pfunc.attrs.get("in_names")

{mplang_nightly-0.1.dev151 → mplang_nightly-0.1.dev152}/mplang/ops/base.py

@@ -30,7 +30,7 @@ from mplang.core.tensor import TensorType
 # The standard return contract for frontend operations (FeOperation.trace).
 #
 # Triad := (PFunction, list[MPObject], PyTreeDef)
-# - PFunction: Captures fn_type (routing key, e.g., "mlir.stablehlo", "sql.duckdb"),
+# - PFunction: Captures fn_type (routing key, e.g., "mlir.stablehlo", "sql.run"),
 #              input/output MPTypes and optional attributes.
 # - list[MPObject]: The flat positional MPObjects captured under the current
 #                   context (Trace/Interp). Order matches pfunc.ins_info.
@@ -422,6 +422,6 @@ def list_ops(module: str | None = None) -> dict[tuple[str, str], FeOperation]:
 # - For type-only kernels, use @module.typed_op(pfunc_name)(kernel). The op name is derived from the kernel function name.
 # - For complex ops (with Python callables/closures), subclass FeOperation and register
 #   using get_registry().register_op(module, name, op_instance) or use @module.feop with InlineFeOperation.
-# - Ensure PFunction.fn_type is set as the routing key (e.g., "mlir.stablehlo", "sql.duckdb").
+# - Ensure PFunction.fn_type is set as the routing key (e.g., "mlir.stablehlo", "sql.run").
 # - Keep device selection/routing out of frontend code; only set fn_type and attributes.
 # - Avoid moving MPObjects across contexts directly; capture within current ctx in trace().

{mplang_nightly-0.1.dev151 → mplang_nightly-0.1.dev152}/mplang/ops/ibis_cc.py

@@ -65,6 +65,7 @@ def ibis2sql(
         ins_info=tuple(ins_info),
         outs_info=tuple(outs_info),
         in_names=tuple(in_names),
+        dialect="duckdb",
     )
     return pfn
 

{mplang_nightly-0.1.dev151 → mplang_nightly-0.1.dev152}/mplang/ops/sql.py

@@ -46,12 +46,13 @@ class SqlFE(FeOperation):
                 in_vars.append(tbl)
 
         pfn = PFunction(
-            fn_type=f"sql[{self._dialect}]",
+            fn_type="sql.run",
             fn_name="",
             fn_text=sql,
             ins_info=tuple(ins_info),
             outs_info=(out_type,),
             in_names=tuple(in_names),
+            dialect=self._dialect,
         )
         _, treedef = tree_flatten(out_type)
         return pfn, in_vars, treedef