Fix

xorbitsai · Sep 16, 2023 · 87ec47d · 87ec47d
1 parent 994fbff
commit 87ec47d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/xorbits/_mars/dataframe/merge/merge.py b/python/xorbits/_mars/dataframe/merge/merge.py
@@ -68,7 +68,7 @@
 cudf = lazy_import("cudf")
 
 
-class DataFrameMergeAlign(MapReduceOperand, DataFrameAutoMergeMixin):
+class DataFrameMergeAlign(MapReduceOperand, DataFrameOperandMixin):
     _op_type_ = OperandDef.DATAFRAME_SHUFFLE_MERGE_ALIGN
 
     index_shuffle_size = Int32Field("index_shuffle_size")
@@ -174,7 +174,7 @@ class MergeMethod(Enum):
     shuffle = 2
 
 
-class DataFrameMerge(DataFrameOperand, DataFrameOperandMixin):
+class DataFrameMerge(DataFrameOperand, DataFrameAutoMergeMixin):
     _op_type_ = OperandDef.DATAFRAME_MERGE
 
     how = StringField("how")