def get_queries()

in kbc/models.py [0:0]

• 2 lines of code
• 1 McCabe index (conditional complexity)

    def get_queries(self, queries: torch.Tensor):
        return self.lhs(queries[:, 0]).data * self.rel(queries[:, 1]).data