Skip to content

Commit f388a4c

Browse files
committed
Correct class names and internal attr
1 parent 61b6a69 commit f388a4c

File tree

2 files changed

+9
-9
lines changed

2 files changed

+9
-9
lines changed

python/datafusion/context.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -810,11 +810,11 @@ def register_dataset(self, name: str, dataset: pyarrow.dataset.Dataset) -> None:
810810

811811
def register_udf(self, udf: ScalarUDF) -> None:
812812
"""Register a user-defined function (UDF) with the context."""
813-
self.ctx.register_udf(udf.udf)
813+
self.ctx.register_udf(udf._udf)
814814

815815
def register_udaf(self, udaf: AggregateUDF) -> None:
816816
"""Register a user-defined aggregation function (UDAF) with the context."""
817-
self.ctx.register_udaf(udaf)
817+
self.ctx.register_udaf(udaf._udaf)
818818

819819
def catalog(self, name: str = "datafusion") -> Catalog:
820820
"""Retrieve a catalog by name."""

python/datafusion/substrait.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def serialize(sql: str, ctx: SessionContext, path: str | pathlib.Path) -> None:
7474
ctx: SessionContext to use.
7575
path: Path to write the Substrait plan to.
7676
"""
77-
return substrait_internal.serde.serialize(sql, ctx.ctx, str(path))
77+
return substrait_internal.Serde.serialize(sql, ctx.ctx, str(path))
7878

7979
@staticmethod
8080
def serialize_to_plan(sql: str, ctx: SessionContext) -> Plan:
@@ -87,7 +87,7 @@ def serialize_to_plan(sql: str, ctx: SessionContext) -> Plan:
8787
Returns:
8888
Substrait plan.
8989
"""
90-
return Plan(substrait_internal.serde.serialize_to_plan(sql, ctx.ctx))
90+
return Plan(substrait_internal.Serde.serialize_to_plan(sql, ctx.ctx))
9191

9292
@staticmethod
9393
def serialize_bytes(sql: str, ctx: SessionContext) -> bytes:
@@ -100,7 +100,7 @@ def serialize_bytes(sql: str, ctx: SessionContext) -> bytes:
100100
Returns:
101101
Substrait plan as bytes.
102102
"""
103-
return substrait_internal.serde.serialize_bytes(sql, ctx.ctx)
103+
return substrait_internal.Serde.serialize_bytes(sql, ctx.ctx)
104104

105105
@staticmethod
106106
def deserialize(path: str | pathlib.Path) -> Plan:
@@ -112,7 +112,7 @@ def deserialize(path: str | pathlib.Path) -> Plan:
112112
Returns:
113113
Substrait plan.
114114
"""
115-
return Plan(substrait_internal.serde.deserialize(str(path)))
115+
return Plan(substrait_internal.Serde.deserialize(str(path)))
116116

117117
@staticmethod
118118
def deserialize_bytes(proto_bytes: bytes) -> Plan:
@@ -124,7 +124,7 @@ def deserialize_bytes(proto_bytes: bytes) -> Plan:
124124
Returns:
125125
Substrait plan.
126126
"""
127-
return Plan(substrait_internal.serde.deserialize_bytes(proto_bytes))
127+
return Plan(substrait_internal.Serde.deserialize_bytes(proto_bytes))
128128

129129

130130
@deprecated("Use `Serde` instead.")
@@ -149,7 +149,7 @@ def to_substrait_plan(logical_plan: LogicalPlan, ctx: SessionContext) -> Plan:
149149
Substrait plan.
150150
"""
151151
return Plan(
152-
substrait_internal.producer.to_substrait_plan(logical_plan, ctx.ctx)
152+
substrait_internal.Producer.to_substrait_plan(logical_plan, ctx.ctx)
153153
)
154154

155155

@@ -174,7 +174,7 @@ def from_substrait_plan(ctx: SessionContext, plan: Plan) -> LogicalPlan:
174174
Returns:
175175
LogicalPlan.
176176
"""
177-
return substrait_internal.consumer.from_substrait_plan(
177+
return substrait_internal.Consumer.from_substrait_plan(
178178
ctx.ctx, plan.plan_internal
179179
)
180180

0 commit comments

Comments
 (0)