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

timsaucer · Aug 5, 2024 · 9d7dda2 · 9d7dda2
1 parent dd5e111
commit 9d7dda2
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/python/datafusion/substrait.py b/python/datafusion/substrait.py
@@ -81,7 +81,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:
@@ -94,7 +94,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:
@@ -107,7 +107,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:
@@ -119,7 +119,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:
@@ -131,7 +131,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.")
@@ -156,7 +156,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)
         )
 
 
@@ -181,7 +181,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
         )