Method load_state_dict

pointersect/inference/structures.py:260–267

Load the state dictionary.

(
            self,
            state_dict: T.Dict[str, T.Any],
    )

        return to_save

    def load_state_dict(
            self,
            state_dict: T.Dict[str, T.Any],
    ):
        """Load the state dictionary."""
        for name in self.attr_names:
            setattr(self, name, state_dict.get(name, None))
        setattr(self, 'included_point_at_inf', state_dict.get('included_point_at_inf', False))

    def extract_valid_attr(self, arr: torch.Tensor, bidx: int) -> T.Union[torch.Tensor, None]:
        """