def __sub__()

in torcharrow/velox_rt/dataframe_cpu.py [0:0]

• 35 lines of code
• 4 McCabe index (conditional complexity)

    def __sub__(self, other):
        if isinstance(other, DataFrameCpu):
            assert len(self) == len(other)
            return self._fromdata(
                {
                    self.dtype.fields[i].name: ColumnFromVelox._from_velox(
                        self.device,
                        self.dtype.fields[i].dtype,
                        self._data.child_at(i),
                        True,
                    )
                    - ColumnFromVelox._from_velox(
                        other.device,
                        other.dtype.fields[i].dtype,
                        other._data.child_at(i),
                        True,
                    )
                    for i in range(self._data.children_size())
                },
                self._mask,
            )
        else:
            return self._fromdata(
                {
                    self.dtype.fields[i].name: ColumnFromVelox._from_velox(
                        self.device,
                        self.dtype.fields[i].dtype,
                        self._data.child_at(i),
                        True,
                    )
                    - other
                    for i in range(self._data.children_size())
                },
                self._mask,
            )