Another attempt to fix vector doc type problem

markgw · Feb 10, 2020 · 6920956 · 6920956
1 parent a0bb3d2
commit 6920956
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/python/pimlico/datatypes/corpora/floats.py b/src/python/pimlico/datatypes/corpora/floats.py
@@ -195,14 +195,14 @@ class VectorDocumentType(RawDocumentType):
     def reader_init(self, reader):
         super(VectorDocumentType, self).reader_init(reader)
         # Prepare struct for reading in the whole vector
-        dim = reader.datatype.metadata["dimensions"]
+        dim = reader.metadata["dimensions"]
         self.struct = struct.Struct("<" + "d"*dim)
         self.value_size = self.struct.size
 
     def writer_init(self, writer):
         super(VectorDocumentType, self).writer_init(writer)
         # Prepare struct for writing out the whole vector
-        dim = writer.datatype.metadata["dimensions"]
+        dim = writer.metadata["dimensions"]
         self.struct = struct.Struct("<" + "d"*dim)
         self.value_size = self.struct.size