Still trying to fix vector datatype problem

markgw · Feb 10, 2020 · a0bb3d2 · a0bb3d2
1 parent b30fa24
commit a0bb3d2
Showing 1 changed file with 4 additions and 4 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 = self.metadata["dimensions"]
+        dim = reader.datatype.metadata["dimensions"]
         self.struct = struct.Struct("<" + "d"*dim)
         self.value_size = self.struct.size
 
-    def writer_init(self, reader):
-        super(VectorDocumentType, self).writer_init(reader)
+    def writer_init(self, writer):
+        super(VectorDocumentType, self).writer_init(writer)
         # Prepare struct for writing out the whole vector
-        dim = self.metadata["dimensions"]
+        dim = writer.datatype.metadata["dimensions"]
         self.struct = struct.Struct("<" + "d"*dim)
         self.value_size = self.struct.size