From 7396d085fe64fc8962674b722667f5c87c4a8155 Mon Sep 17 00:00:00 2001 From: Simon Willison Date: Thu, 17 Aug 2023 16:54:38 -0700 Subject: [PATCH] Fixed two mypy errors --- sqlite_utils/db.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/sqlite_utils/db.py b/sqlite_utils/db.py index 534d2ee..7d39696 100644 --- a/sqlite_utils/db.py +++ b/sqlite_utils/db.py @@ -778,10 +778,10 @@ class Database: assert ( tuple_or_list[0] == name ), "First item in {} should have been {}".format(tuple_or_list, name) - tuple_or_list = tuple(tuple_or_list[1:]) assert len(tuple_or_list) in ( 2, 3, + 4, ), "foreign_keys= should be a list of tuple pairs or triples" if len(tuple_or_list) == 3: tuple_or_list = cast(Tuple[str, str, str], tuple_or_list) @@ -1161,12 +1161,12 @@ class Database: ) # Group them by table - by_table = {} + by_table: Dict[str, List] = {} for fk in foreign_keys_to_create: by_table.setdefault(fk[0], []).append(fk) for table, fks in by_table.items(): - self[table].transform(add_foreign_keys=fks) + cast(Table, self[table]).transform(add_foreign_keys=fks) self.vacuum()