| 939 | |
| 940 | |
| 941 | class MapOverlap(MapPartitions): |
| 942 | _parameters = [ |
| 943 | "frame", |
| 944 | "func", |
| 945 | "before", |
| 946 | "after", |
| 947 | "meta", |
| 948 | "enforce_metadata", |
| 949 | "transform_divisions", |
| 950 | "clear_divisions", |
| 951 | "align_dataframes", |
| 952 | "token", |
| 953 | "kwargs", |
| 954 | ] |
| 955 | _defaults: dict = { |
| 956 | "meta": None, |
| 957 | "enfore_metadata": True, |
| 958 | "transform_divisions": True, |
| 959 | "kwargs": None, |
| 960 | "clear_divisions": False, |
| 961 | "align_dataframes": False, |
| 962 | "token": None, |
| 963 | } |
| 964 | |
| 965 | @functools.cached_property |
| 966 | def _kwargs(self) -> dict: |
| 967 | kwargs = self.kwargs |
| 968 | if kwargs is None: |
| 969 | kwargs = {} |
| 970 | return kwargs |
| 971 | |
| 972 | @property |
| 973 | def args(self): |
| 974 | return ( |
| 975 | [self.frame] |
| 976 | + [self.func, self.before, self.after] |
| 977 | + self.operands[len(self._parameters) :] |
| 978 | ) |
| 979 | |
| 980 | @functools.cached_property |
| 981 | def _meta(self): |
| 982 | meta = self.operand("meta") |
| 983 | args = [self.frame._meta] + [ |
| 984 | arg._meta if isinstance(arg, Expr) else arg |
| 985 | for arg in self.operands[len(self._parameters) :] |
| 986 | ] |
| 987 | return _get_meta_map_partitions( |
| 988 | args, |
| 989 | [self.dependencies()[0]], |
| 990 | self.func, |
| 991 | self.kwargs, |
| 992 | meta, |
| 993 | self.kwargs.pop("parent_meta", None), |
| 994 | ) |
| 995 | |
| 996 | @functools.cached_property |
| 997 | def before(self): |
| 998 | before = self.operand("before") |