8d0f7cb1babdbcedf7669f2dfddc88cd4c030167,test/experimental/test_vectors.py,TestVectors,test_vectors_load_and_save,#TestVectors#,100

Before Change

        expected_unk_tensor = torch.tensor([0, 0], dtype=torch.float)

        tokens = ["a"]
        vectors = tensorA.unsqueeze(0)
        vectors_obj = Vectors(tokens, vectors)

        vector_path = os.path.join(self.test_dir, "vectors.pt")
        torch.save(vectors_obj, vector_path)

After Change

    def test_vectors_load_and_save(self):
        tensorA = torch.tensor([1, 0], dtype=torch.float)
        tensorB = torch.tensor([0, 1], dtype=torch.float)
        expected_unk_tensor = torch.tensor([0, 0], dtype=torch.float)

        tokens = ["a", "b"]
        vectors = torch.stack((tensorA, tensorB), 0)

Instances