QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-06-01 21:59:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: upgrade mypy and add type guards (#16227)

Browse Source
This commit is contained in:
Ville Brofeldt
2021-08-14 06:31:45 +03:00
committed by GitHub
parent 9b2dffeb1d
commit d46dc9aa45
9 changed files with 29 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/benchmark_migration.py
View File

@@ -44,9 +44,13 @@ def import_migration_script(filepath: Path) -> ModuleType:
Import migration script as if it were a module.
"""
spec = importlib.util.spec_from_file_location(filepath.stem, filepath)
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module) # type: ignore
return module
if spec:
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module) # type: ignore
return module
raise Exception(
"No module spec found in location: `{path}`".format(path=str(filepath))
)
def extract_modified_tables(module: ModuleType) -> Set[str]: