@@ -2738,14 +2738,13 @@ def needs_migrate_impl(
27382738 return DataMigrationStatus (needs_migrate = False , migration_done = True )
27392739
27402740
2741- class EntityIDPostMigration (BaseMigrationWithQuery , BaseRunTimeMigration ):
2741+ class EntityIDPostMigration (BaseMigrationWithQuery , BaseOffLineMigration ):
27422742 """Migration to remove old entity_id strings from states.
27432743
27442744 Introduced in HA Core 2023.4 by PR #89557.
27452745 """
27462746
27472747 migration_id = "entity_id_post_migration"
2748- task = MigrationTask
27492748 index_to_drop = (TABLE_STATES , LEGACY_STATES_ENTITY_ID_LAST_UPDATED_INDEX )
27502749
27512750 def migrate_data_impl (self , instance : Recorder ) -> DataMigrationStatus :
@@ -2758,16 +2757,16 @@ def needs_migrate_query(self) -> StatementLambdaElement:
27582757 return has_used_states_entity_ids ()
27592758
27602759
2761- NON_LIVE_DATA_MIGRATORS = (
2760+ NON_LIVE_DATA_MIGRATORS : tuple [ type [ BaseOffLineMigration ], ...] = (
27622761 StatesContextIDMigration , # Introduced in HA Core 2023.4
27632762 EventsContextIDMigration , # Introduced in HA Core 2023.4
27642763 EventTypeIDMigration , # Introduced in HA Core 2023.4 by PR #89465
27652764 EntityIDMigration , # Introduced in HA Core 2023.4 by PR #89557
2765+ EntityIDPostMigration , # Introduced in HA Core 2023.4 by PR #89557
27662766)
27672767
2768- LIVE_DATA_MIGRATORS = (
2768+ LIVE_DATA_MIGRATORS : tuple [ type [ BaseRunTimeMigration ], ...] = (
27692769 EventIDPostMigration , # Introduced in HA Core 2023.4 by PR #89901
2770- EntityIDPostMigration , # Introduced in HA Core 2023.4 by PR #89557
27712770)
27722771
27732772
0 commit comments