fix: broken glue extractor when there is a description (#55)

rsyi · Oct 23, 2020 · 18425da · 18425da
1 parent 189b8f9
commit 18425da
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pipelines/setup.py b/pipelines/setup.py
@@ -8,7 +8,7 @@
 
 setuptools.setup(
     name='whale-pipelines',
-    version='1.1.0',
+    version='1.1.1',
     author='Robert Yi',
     author_email='[email protected]',
     description="A pared-down metadata scraper + SQL runner.",

diff --git a/pipelines/whale/transformer/markdown_transformer.py b/pipelines/whale/transformer/markdown_transformer.py
@@ -6,6 +6,7 @@
     COLUMN_DETAILS_DELIMITER
 )
 from databuilder.models.watermark import Watermark
+from databuilder.models.table_metadata import DescriptionMetadata
 
 import textwrap
 
@@ -26,7 +27,10 @@ def format_table_metadata(
         formatted_columns = self.format_columns(record)
 
         if record.description:
-            description = record.description + "\n"
+            if type(record.description) == DescriptionMetadata:
+                description = record.description._text + "\n"
+            else:
+                description = str(record.description) + "\n"
         else:
             description = ""