diff --git a/superset/cli/viz_migrations.py b/superset/cli/viz_migrations.py index 8b8484bdf4522..c80c3a5136945 100644 --- a/superset/cli/viz_migrations.py +++ b/superset/cli/viz_migrations.py @@ -27,8 +27,8 @@ class VizType(str, Enum): AREA = "area" BUBBLE = "bubble" DUAL_LINE = "dual_line" - LINE = "line" HEATMAP = "heatmap" + LINE = "line" PIVOT_TABLE = "pivot_table" SUNBURST = "sunburst" TREEMAP = "treemap" @@ -80,8 +80,8 @@ def migrate(viz_type: VizType, is_downgrade: bool = False) -> None: MigrateAreaChart, MigrateBubbleChart, MigrateDualLine, - MigrateLineChart, MigrateHeatmapChart, + MigrateLineChart, MigratePivotTable, MigrateSunburst, MigrateTreeMap, @@ -91,8 +91,8 @@ def migrate(viz_type: VizType, is_downgrade: bool = False) -> None: VizType.AREA: MigrateAreaChart, VizType.BUBBLE: MigrateBubbleChart, VizType.DUAL_LINE: MigrateDualLine, - VizType.LINE: MigrateLineChart, VizType.HEATMAP: MigrateHeatmapChart, + VizType.LINE: MigrateLineChart, VizType.PIVOT_TABLE: MigratePivotTable, VizType.SUNBURST: MigrateSunburst, VizType.TREEMAP: MigrateTreeMap,