Add guard for token sniffer score

tradingstrategy-ai · Feb 11, 2025 · 53c53a6 · 53c53a6
1 parent 9327e2e
commit 53c53a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tradingstrategy/token_metadata.py b/tradingstrategy/token_metadata.py
@@ -86,7 +86,7 @@ def token_sniffer_score(self) -> int | None:
         if self.token_sniffer_data is None:
             return None
 
-        return self.token_sniffer_data["score"]
+        return self.token_sniffer_data.get("score")
 
     def get_coingecko_categories(self) -> set[str] | None:
         """Get CoinGecko categories of this token.

diff --git a/tradingstrategy/top.py b/tradingstrategy/top.py
@@ -187,7 +187,7 @@ def token_sniffer_score(self) -> int | None:
         if self.token_sniffer_data is None:
             return None
 
-        return self.token_sniffer_data["score"]
+        return self.token_sniffer_data.get("score")
 
     def has_tax_data(self) -> bool | None:
         """Do we have tax data for this pair.