Another fix

markgw · Feb 10, 2020 · a0be78d · a0be78d
1 parent 2682972
commit a0be78d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/python/pimlico/core/modules/map/__init__.py b/src/python/pimlico/core/modules/map/__init__.py
@@ -304,7 +304,8 @@ def map_documents(self):
 
         # Get the expected output datatypes, ready for any possible output type conversion when we get results
         output_datatypes = [
-            executor.info.get_output_datatype(name)[1] for name in executor.info.get_grouped_corpus_output_names()
+            executor.info.get_output_datatype(name)[1].data_point_type
+            for name in executor.info.get_grouped_corpus_output_names()
         ]
         num_outputs = len(output_datatypes)