Skip to content

Commit 1d9ce46

Browse files
authored
Fix mypy assertion error in upgrade_0_to_1 (#111)
Add type ignore comments for locally-scoped UpdatedBase and its subclasses to resolve 'Cannot find module for UpdatedBase' error.
1 parent ed38846 commit 1d9ce46

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

lnt/server/db/migrations/upgrade_0_to_1.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -178,10 +178,10 @@ def initialize_compile_definition(session):
178178

179179

180180
def get_base_for_testsuite(test_suite):
181-
UpdatedBase = declarative_base()
181+
UpdatedBase = declarative_base() # type: ignore[misc]
182182
db_key_name = test_suite.db_key_name
183183

184-
class Machine(UpdatedBase):
184+
class Machine(UpdatedBase): # type: ignore[misc,valid-type]
185185
__tablename__ = db_key_name + '_Machine'
186186
# For case sensitive compare.
187187
__table_args__ = {'mysql_collate': 'utf8_bin'}
@@ -200,7 +200,7 @@ class Machine(UpdatedBase):
200200
class_dict[item.name] = item.column = Column(
201201
item.name, String(256))
202202

203-
class Order(UpdatedBase):
203+
class Order(UpdatedBase): # type: ignore[misc,valid-type]
204204
__tablename__ = db_key_name + '_Order'
205205

206206
id = Column("ID", Integer, primary_key=True)
@@ -219,7 +219,7 @@ class Order(UpdatedBase):
219219
class_dict[item.name] = item.column = Column(
220220
item.name, String(256))
221221

222-
class Run(UpdatedBase):
222+
class Run(UpdatedBase): # type: ignore[misc,valid-type]
223223
__tablename__ = db_key_name + '_Run'
224224

225225
id = Column("ID", Integer, primary_key=True)
@@ -247,13 +247,13 @@ class Run(UpdatedBase):
247247
class_dict[item.name] = item.column = Column(
248248
item.name, String(256))
249249

250-
class Test(UpdatedBase):
250+
class Test(UpdatedBase): # type: ignore[misc,valid-type]
251251
__tablename__ = db_key_name + '_Test'
252252
__table_args__ = {'mysql_collate': 'utf8_bin'}
253253
id = Column("ID", Integer, primary_key=True)
254254
name = Column("Name", String(256), unique=True, index=True)
255255

256-
class Sample(UpdatedBase):
256+
class Sample(UpdatedBase): # type: ignore[misc,valid-type]
257257
__tablename__ = db_key_name + '_Sample'
258258

259259
id = Column("ID", Integer, primary_key=True)

0 commit comments

Comments
 (0)