def forward()

in ember/create.py [0:0]

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

    def forward(self, input_ids, attention_mask):
        outputs = self.model.forward(
            {"input_ids": input_ids, "attention_mask": attention_mask}
        )
        return outputs["sentence_embedding"]