def update_data()

in theseus/embodied/collision/signed_distance_field.py [0:0]

• 23 lines of code
• 10 McCabe index (conditional complexity)

    def update_data(self, origin: Variable, sdf_data: Variable, cell_size: Variable):
        if sdf_data.ndim != 3:
            raise ValueError(
                "Argument sdf_data to SignedDistanceField2D must be a batch of matrices."
            )
        if not (origin.ndim == 2 or (origin.ndim == 3 and origin.shape[2] == 1)):
            raise ValueError(
                "Argument origin to SignedDistanceField2D must be a batch of 2-D tensors."
            )
        if not (
            cell_size.ndim == 1 or (cell_size.ndim == 2 and cell_size.shape[1] == 1)
        ):
            raise ValueError(
                "Argument cell_size must be a batch of 0-D or 1-D tensors."
            )
        if (
            origin.shape[0] != sdf_data.shape[0]
            or origin.shape[0] != cell_size.shape[0]
        ):
            raise ValueError("Incompatible batch size between input arguments.")
        self.origin = origin
        self.sdf_data = sdf_data
        self.cell_size = cell_size