Fix a crash if WSS/RRF values are NA (#2)

asreview · Mar 12, 2020 · 388e226 · 388e226
1 parent fee9485
commit 388e226
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/asreviewcontrib/statistics/__init__.py b/asreviewcontrib/statistics/__init__.py
@@ -16,4 +16,4 @@
 from asreviewcontrib.statistics.statistics import DataStatistics
 from asreviewcontrib.statistics.entrypoint import StatEntryPoint
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
diff --git a/asreviewcontrib/statistics/statistics.py b/asreviewcontrib/statistics/statistics.py
@@ -96,10 +96,16 @@ def __str__(self):
         if len(results["wss"]) + len(results["rrf"]) > 0:
             stat_str += "-----------  WSS/RRF  -----------\n"
             for wss_at, wss_val in results["wss"].items():
-                wss_val_str = f"{wss_val:.2f}"
+                if wss_val is not None:
+                    wss_val_str = f"{wss_val:.2f}"
+                else:
+                    wss_val_str = "NA"
                 stat_str += f"WSS@{wss_at: <3}: {wss_val_str: <5} %\n"
             for rrf_at, rrf_val in results["rrf"].items():
-                rrf_val_str = f"{rrf_val:.2f}"
+                if rrf_val is not None:
+                    rrf_val_str = f"{rrf_val:.2f}"
+                else:
+                    rrf_val_str = "NA"
                 stat_str += f"RRF@{rrf_at: <3}: {rrf_val_str: <5} %\n"
 
         return stat_str