mplang-nightly 0.1.dev151__py3-none-any.whl → 0.1.dev152__py3-none-any.whl

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/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/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/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

{mplang_nightly-0.1.dev151.dist-info → mplang_nightly-0.1.dev152.dist-info}/METADATA

@@ -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.dist-info → mplang_nightly-0.1.dev152.dist-info}/RECORD

@@ -34,17 +34,17 @@ mplang/kernels/crypto.py,sha256=TWixli1uRQ_7OjA49qQXUXa2ldHDEwaCMXXPSHdAPi8,3812
 mplang/kernels/mock_tee.py,sha256=ifVqb03FgQ5EDK8r14PMpZKMMaSeqMDorcpGLiy00mM,2233
 mplang/kernels/phe.py,sha256=8-_1IFPOaGECGj9mbYja8XoqbMYnYqfpDNVyMJd8J1Y,65247
 mplang/kernels/spu.py,sha256=Kkg1ZQbmklXl7YkIeBfxqs3o4wX7ygBE8hXLpx90L9Y,9307
-mplang/kernels/sql_duckdb.py,sha256=C2XdNLoE2Apma-Fs7OYzDzkBAXAVuShuROxhCWCHDG4,1502
+mplang/kernels/sql_duckdb.py,sha256=UN1Ev6-MxF_-65zMExUsLScC9PlmEIEcN8YziIoX_rY,1724
 mplang/kernels/stablehlo.py,sha256=jDsu-lIHRAm94FcUcxZgK02c6BhFJpbO8cf2hP2DFgk,2937
 mplang/ops/__init__.py,sha256=dpe7WWiYapOFzJeGoKFYBr5mnd6P5SdOyvdYaM2Nhm0,1408
-mplang/ops/base.py,sha256=rGtfBejcDh9mTRxOdJK5VUlG5vYiVJSir8X72X0Huvc,18264
+mplang/ops/base.py,sha256=h67_SHWNZGUuTCuMll-9kDgGvlPhlFov7WAQCHTmUvw,18258
 mplang/ops/builtin.py,sha256=D7T8rRF9g05VIw9T72lsncF5cDQqaT37eapBieRKvRI,9363
 mplang/ops/crypto.py,sha256=9CeFJrYmvjmgx-3WQl6jHXh8VafRpT4QBunbzsPF8Uc,3646
-mplang/ops/ibis_cc.py,sha256=bWKN1dL8Nluwvu5TLi8iUwytcnpXtWakZDCL793zBRA,4230
+mplang/ops/ibis_cc.py,sha256=G8TobBlkyT_IxOWXmMm4APPUnj6sH9bJ-1STfw4kIcw,4256
 mplang/ops/jax_cc.py,sha256=42czYg3hNQbI_nUebXnshlU8ULwM-oBDe_TQoApLNVA,7802
 mplang/ops/phe.py,sha256=SatswExjZWPed8y3qA33BCwIWbvsgHCuCAz_pv2RLLw,6790
 mplang/ops/spu.py,sha256=UHr5DSoqG08xDYER_11OsMVjGGNXXxsvkFoVvXU8uik,4989
-mplang/ops/sql.py,sha256=p-u0wQPk9KlgveltYvQcF1UefScJoqBCqhzYPeLBB5Y,1994
+mplang/ops/sql.py,sha256=HyY2i5aGC5W7r62JryFSjQCUDXH3kQz82YADwn4z5uc,2015
 mplang/ops/tee.py,sha256=gwzP81y2idH-d-Du84H6oNZpLaGD-3fEgm8G1uxWpUA,1388
 mplang/protos/v1alpha1/mpir_pb2.py,sha256=Bros37t-4LMJbuUYVSM65rImUYTtZDhNTIADGbZCKp0,7522
 mplang/protos/v1alpha1/mpir_pb2.pyi,sha256=GwXR4wPB_kB_36iYS9x-cGI9KDKFMq89KhdLhW_xmvE,19342
@@ -70,8 +70,8 @@ mplang/utils/crypto.py,sha256=rvPomBFtznRHc3RPi6Aip9lsU8zW2oxBqGv1K3vn7Rs,1052
 mplang/utils/func_utils.py,sha256=vCJcZmu0bEbqhOQKdpttV2_MBllIcPSN0b8U4WjNGGo,5164
 mplang/utils/spu_utils.py,sha256=S3L9RBkBe2AvSuMSQQ12cBY5Y1NPthubvErSX_7nj1A,4158
 mplang/utils/table_utils.py,sha256=aC-IZOKkSmFkpr3NZchLM0Wt0GOn-rg_xHBHREWBwAU,2202
-mplang_nightly-0.1.dev151.dist-info/METADATA,sha256=sCQECTJOQoKyr3XXAf8Kma7lrB5KEt6toJbQA9a5nEA,16547
-mplang_nightly-0.1.dev151.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mplang_nightly-0.1.dev151.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
-mplang_nightly-0.1.dev151.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-mplang_nightly-0.1.dev151.dist-info/RECORD,,
+mplang_nightly-0.1.dev152.dist-info/METADATA,sha256=5Mt-R98IyuopKWLvWdxbCio3Yo8KmmmYqs92FyETS5M,16547
+mplang_nightly-0.1.dev152.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mplang_nightly-0.1.dev152.dist-info/entry_points.txt,sha256=mG1oJT-GAjQR834a62_QIWb7litzWPPyVnwFqm-rWuY,55
+mplang_nightly-0.1.dev152.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+mplang_nightly-0.1.dev152.dist-info/RECORD,,