Fixup type annotation

nsidc · Oct 2, 2024 · f2d508d · f2d508d
1 parent 2e31159
commit f2d508d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/usaon_benefit_tool/util/sankey.py b/usaon_benefit_tool/util/sankey.py
@@ -180,7 +180,7 @@ def _node_ids_in_links(links: list[HighchartsSankeySeriesLink]) -> set[str]:
     return node_ids
 
 
-def _weight_for_criticality_rating(criticality_rating: int | None) -> float:
+def _weight_for_criticality_rating(criticality_rating: int | None) -> int | float:
     """If criticality rating is not set, return a very thin line."""
     if criticality_rating is None:
         return 0.01