Fixed issue with topic_names_ attribute when creating tables

x-tabdeveloping · Nov 4, 2024 · 69bdb70 · 69bdb70
1 parent 2d57c48
commit 69bdb70
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/turftopic/base.py b/turftopic/base.py
@@ -107,7 +107,7 @@ def _topics_table(
         show_negative: bool = False,
     ) -> list[list[str]]:
         columns = ["Topic ID"]
-        if hasattr(self, "topic_names_"):
+        if getattr(self, "topic_names_", None):
             columns.append("Topic Name")
         columns.append("Highest Ranking")
         if show_negative:
@@ -146,7 +146,7 @@ def _topics_table(
                 concat_positive = ", ".join([word for word in vocab[highest]])
                 concat_negative = ", ".join([word for word in vocab[lowest]])
             row = [f"{topic_id}"]
-            if hasattr(self, "topic_names_"):
+            if getattr(self, "topic_names_", None):
                 row.append(self.topic_names_[i_topic])
             row.append(f"{concat_positive}")
             if show_negative: