46b03dae5a8333a38c072da8e541da6fe5c0912d,grid/services/torch/torch_service.py,TorchService,receive_obj_break,#TorchService#Any#,36

Before Change

                // Ensure
                tensor_type = tu.types_guard(_tensor_type)
            except KeyError:
                raise TypeError(
                    "Tried to receive a non-Torch object of type {}.".format(
                        _tensor_type))
            // this could be a significant failure point, security-wise
            if ("data" in msg.keys()):
                data = obj_msg["data"]
                data = tu.tensor_contents_guard(data)

After Change

        try:
            torch_type = tu.types_guard(obj_msg)

            if torch_type in self.var_types:
                v = self.build_var(obj_msg, torch_type)
            else:
                v = self.build_tensor(obj_msg, torch_type)

            return self.handle_register(v, obj_msg)

Instances