ENH: strip some formatted numbers to handle NANs

jeffgortmaker · Aug 23, 2022 · 7de390c · 7de390c
1 parent 55b7b47
commit 7de390c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/pyblp/economies/problem.py b/pyblp/economies/problem.py
@@ -860,7 +860,10 @@ def market_factory(
                         if micro_value.size != 1:
                             raise TypeError(f"compute_value of micro moment '{moment}' should return a float.")
                         if not np.isfinite(micro_value):
-                            warn(f"compute_value of micro moment '{moment}' returned {format_number(micro_value)}.")
+                            warn(
+                                f"compute_value of micro moment '{moment}' returned "
+                                f"{format_number(micro_value).strip()}."
+                            )
 
                         micro_values[m] = micro_value
 
@@ -876,7 +879,7 @@ def market_factory(
                                 if not np.isfinite(micro_gradient[p]):
                                     warn(
                                         f"compute_gradient of micro moment '{moment}' returned "
-                                        f"{format_number(micro_gradient[p])} for part '{part}'."
+                                        f"{format_number(micro_gradient[p]).strip()} for part '{part}'."
                                     )
 
                             micro_gradients[m, part_indices] = micro_gradient

diff --git a/pyblp/utilities/basics.py b/pyblp/utilities/basics.py
@@ -601,7 +601,7 @@ def __init__(self, matrix: Array) -> None:
 
     def __str__(self) -> str:
         """Supplement the error with the condition number."""
-        return f"{super().__str__()} Condition number: {format_number(self._condition)}."
+        return f"{super().__str__()} Condition number: {format_number(self._condition).strip()}."
 
 
 class InversionReplacementError(InversionError):