Catching the index error for unit with no message data

parlai/crowdsourcing/utils/analysis.py

@@ -7,6 +7,7 @@
 from __future__ import annotations
 
 import argparse
+import logging
 import os
 from abc import ABC, abstractmethod
 from datetime import datetime
@@ -185,5 +186,11 @@ def get_units_data(self, task_units: List[Unit]) -> List[dict]:
         data_browser = self.get_mephisto_data_browser()
         task_data = []
         for unit in task_units:
-            task_data.append(data_browser.get_data_from_unit(unit))
+            try:
+                unit_data = data_browser.get_data_from_unit(unit)
+                task_data.append(unit_data)
+            except IndexError:
+                logging.warning(
+                    f"Skipping unit {unit.db_id}. No message found for this unit."
+                )
         return task_data