laser embedding does not return the embeddings of more than two items

[msharaf315]

class Laser_embedding(nn.Module):
def init(self):
super().init()
self.out_features = 1024
self.laser = Laser()
def forward(self, input):
print(input)
return self.laser.embed_sentences(sentences= input,lang=["ar","eng"])

laser = Laser_embedding()
list = ["go","fast","test"]
ouput_embed = np.array(laser(list))
print(ouput_embed.shape)

output is:
['go', 'fast', 'test']
(2, 1024)