@@ -81,7 +81,7 @@ def serialize(sql: str, ctx: SessionContext, path: str | pathlib.Path) -> None:
81
81
ctx: SessionContext to use.
82
82
path: Path to write the Substrait plan to.
83
83
"""
84
- return substrait_internal .Serde .serialize (sql , ctx .ctx , str (path ))
84
+ return substrait_internal .serde .serialize (sql , ctx .ctx , str (path ))
85
85
86
86
@staticmethod
87
87
def serialize_to_plan (sql : str , ctx : SessionContext ) -> Plan :
@@ -94,7 +94,7 @@ def serialize_to_plan(sql: str, ctx: SessionContext) -> Plan:
94
94
Returns:
95
95
Substrait plan.
96
96
"""
97
- return Plan (substrait_internal .Serde .serialize_to_plan (sql , ctx .ctx ))
97
+ return Plan (substrait_internal .serde .serialize_to_plan (sql , ctx .ctx ))
98
98
99
99
@staticmethod
100
100
def serialize_bytes (sql : str , ctx : SessionContext ) -> bytes :
@@ -107,7 +107,7 @@ def serialize_bytes(sql: str, ctx: SessionContext) -> bytes:
107
107
Returns:
108
108
Substrait plan as bytes.
109
109
"""
110
- return substrait_internal .Serde .serialize_bytes (sql , ctx .ctx )
110
+ return substrait_internal .serde .serialize_bytes (sql , ctx .ctx )
111
111
112
112
@staticmethod
113
113
def deserialize (path : str | pathlib .Path ) -> Plan :
@@ -119,7 +119,7 @@ def deserialize(path: str | pathlib.Path) -> Plan:
119
119
Returns:
120
120
Substrait plan.
121
121
"""
122
- return Plan (substrait_internal .Serde .deserialize (str (path )))
122
+ return Plan (substrait_internal .serde .deserialize (str (path )))
123
123
124
124
@staticmethod
125
125
def deserialize_bytes (proto_bytes : bytes ) -> Plan :
@@ -131,7 +131,7 @@ def deserialize_bytes(proto_bytes: bytes) -> Plan:
131
131
Returns:
132
132
Substrait plan.
133
133
"""
134
- return Plan (substrait_internal .Serde .deserialize_bytes (proto_bytes ))
134
+ return Plan (substrait_internal .serde .deserialize_bytes (proto_bytes ))
135
135
136
136
137
137
@deprecated ("Use `Serde` instead." )
@@ -156,7 +156,7 @@ def to_substrait_plan(logical_plan: LogicalPlan, ctx: SessionContext) -> Plan:
156
156
Substrait plan.
157
157
"""
158
158
return Plan (
159
- substrait_internal .Producer .to_substrait_plan (logical_plan , ctx .ctx )
159
+ substrait_internal .producer .to_substrait_plan (logical_plan , ctx .ctx )
160
160
)
161
161
162
162
@@ -181,7 +181,7 @@ def from_substrait_plan(ctx: SessionContext, plan: Plan) -> LogicalPlan:
181
181
Returns:
182
182
LogicalPlan.
183
183
"""
184
- return substrait_internal .Consumer .from_substrait_plan (
184
+ return substrait_internal .consumer .from_substrait_plan (
185
185
ctx .ctx , plan .plan_internal
186
186
)
187
187
0 commit comments