Fix flake8 errors

innolitics · Aug 11, 2022 · 97b1d30 · 97b1d30
1 parent 489aa08
commit 97b1d30
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/n2y/blocks.py b/n2y/blocks.py
@@ -372,7 +372,10 @@ def to_pandoc(self):
 class RowBlock(Block):
     def __init__(self, client, notion_data, page, get_children=True):
         super().__init__(client, notion_data, page, get_children)
-        self.cells = [client.wrap_notion_rich_text_array(nc, self) for nc in self.notion_data["cells"]]
+        self.cells = [
+            client.wrap_notion_rich_text_array(nc, self)
+            for nc in self.notion_data["cells"]
+        ]
 
     def to_pandoc(self):
         cells = [Cell(

diff --git a/n2y/rich_text.py b/n2y/rich_text.py
@@ -8,6 +8,7 @@
 )
 
 from n2y.utils import pandoc_ast_to_markdown
+from n2y.notion_mocks import mock_rich_text
 
 
 logger = logging.getLogger(__name__)
@@ -134,7 +135,7 @@ def to_pandoc(self):
 class TextRichText(RichText):
     @classmethod
     def from_plain_text(klass, client, string, block=None):
-        notion_data = mock_notion_rich_text(string)
+        notion_data = mock_rich_text(string)
         return klass(client, notion_data, block)
 
     def __init__(self, client, notion_data, block=None):