aa7176b583f29a61f4fa050d10ba5d57bf908986,test/experimental/test_vectors.py,TestVectors,test_vectors_load_and_save,#TestVectors#,114

Before Change

        vecs = torch.stack((tensorA, tensorB), 0)
        vectors_obj = build_vectors(tokens, vecs)

        tensorC = torch.tensor([1, 1], dtype=torch.float)
        vectors_obj["b"] = tensorC

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

After Change

        vecs = torch.stack((tensorA, tensorB), 0)
        vectors_obj = build_vectors(tokens, vecs)

        with self.subTest("pybind"):
            vector_path = os.path.join(self.test_dir, "vectors_pybind.pt")
            torch.save(vectors_obj, vector_path)
            loaded_vectors_obj = torch.load(vector_path)

            self.assertEqual(loaded_vectors_obj["a"], tensorA)
            self.assertEqual(loaded_vectors_obj["b"], tensorB)
            self.assertEqual(loaded_vectors_obj["not_in_it"], expected_unk_tensor)

        with self.subTest("torchscript"):
            vector_path = os.path.join(self.test_dir, "vectors_torchscript.pt")
            torch.save(vectors_obj.to_ivalue(), vector_path)
            loaded_vectors_obj = torch.load(vector_path)

Instances