Revert changes that will end up coming in via PR #782

apache · Jul 29, 2024 · ac86096 · ac86096
1 parent f388a4c
commit ac86096
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/python/datafusion/substrait.py b/python/datafusion/substrait.py
@@ -74,7 +74,7 @@ def serialize(sql: str, ctx: SessionContext, path: str | pathlib.Path) -> None:
             ctx: SessionContext to use.
             path: Path to write the Substrait plan to.
         """
-        return substrait_internal.Serde.serialize(sql, ctx.ctx, str(path))
+        return substrait_internal.serde.serialize(sql, ctx.ctx, str(path))
 
     @staticmethod
     def serialize_to_plan(sql: str, ctx: SessionContext) -> Plan:
@@ -87,7 +87,7 @@ def serialize_to_plan(sql: str, ctx: SessionContext) -> Plan:
         Returns:
             Substrait plan.
         """
-        return Plan(substrait_internal.Serde.serialize_to_plan(sql, ctx.ctx))
+        return Plan(substrait_internal.serde.serialize_to_plan(sql, ctx.ctx))
 
     @staticmethod
     def serialize_bytes(sql: str, ctx: SessionContext) -> bytes:
@@ -100,7 +100,7 @@ def serialize_bytes(sql: str, ctx: SessionContext) -> bytes:
         Returns:
             Substrait plan as bytes.
         """
-        return substrait_internal.Serde.serialize_bytes(sql, ctx.ctx)
+        return substrait_internal.serde.serialize_bytes(sql, ctx.ctx)
 
     @staticmethod
     def deserialize(path: str | pathlib.Path) -> Plan:
@@ -112,7 +112,7 @@ def deserialize(path: str | pathlib.Path) -> Plan:
         Returns:
             Substrait plan.
         """
-        return Plan(substrait_internal.Serde.deserialize(str(path)))
+        return Plan(substrait_internal.serde.deserialize(str(path)))
 
     @staticmethod
     def deserialize_bytes(proto_bytes: bytes) -> Plan:
@@ -124,7 +124,7 @@ def deserialize_bytes(proto_bytes: bytes) -> Plan:
         Returns:
             Substrait plan.
         """
-        return Plan(substrait_internal.Serde.deserialize_bytes(proto_bytes))
+        return Plan(substrait_internal.serde.deserialize_bytes(proto_bytes))
 
 
 @deprecated("Use `Serde` instead.")
@@ -149,7 +149,7 @@ def to_substrait_plan(logical_plan: LogicalPlan, ctx: SessionContext) -> Plan:
             Substrait plan.
         """
         return Plan(
-            substrait_internal.Producer.to_substrait_plan(logical_plan, ctx.ctx)
+            substrait_internal.producer.to_substrait_plan(logical_plan, ctx.ctx)
         )
 
 
@@ -174,7 +174,7 @@ def from_substrait_plan(ctx: SessionContext, plan: Plan) -> LogicalPlan:
         Returns:
             LogicalPlan.
         """
-        return substrait_internal.Consumer.from_substrait_plan(
+        return substrait_internal.consumer.from_substrait_plan(
             ctx.ctx, plan.plan_internal
         )