Small mypy fixes

calpaterson · Aug 8, 2024 · 3b88f5d · 3b88f5d
1 parent 239b442
commit 3b88f5d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/csvbase/web/schemaorg.py b/csvbase/web/schemaorg.py
@@ -28,8 +28,10 @@ def to_dataset(table: Table) -> Mapping[str, Any]:
     if table.has_caption():
         obj["description"] = table.caption
     # if we knew the table wasn't big we could refer to XLSX here:
+    distribution = []
     for content_type in [ContentType.CSV, ContentType.PARQUET, ContentType.JSON_LINES]:
-        obj["distribution"].append(to_datadownload(table, content_type))
+        distribution.append(to_datadownload(table, content_type))
+    obj["distribution"] = distribution
 
     return obj
 

diff --git a/tests/test_seo.py b/tests/test_seo.py
@@ -61,8 +61,8 @@ def test_schemaorg_dataset(ten_rows):
     actual = schemaorg.to_dataset(ten_rows)
 
     # do the dates this way
-    assert datetime.fromisoformat(actual.pop("dateCreated")) == ten_rows.created
-    assert datetime.fromisoformat(actual.pop("dateModified")) == ten_rows.last_changed
+    assert datetime.fromisoformat(actual.pop("dateCreated")) == ten_rows.created  # type: ignore
+    assert datetime.fromisoformat(actual.pop("dateModified")) == ten_rows.last_changed  # type: ignore
 
     # the rest must match:
     assert expected == actual