clean up

Lightning-AI · Apr 21, 2021 · 792e3b3 · 792e3b3
1 parent 3b44a39
commit 792e3b3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 11 deletions.
diff --git a/pytorch_lightning/utilities/migration/base.py b/pytorch_lightning/utilities/migration/base.py
@@ -113,6 +113,9 @@ def default_migration(checkpoint):
     return checkpoint
 
 
+all_migrations = dict((ver, default_migration) for ver in version_history)
+
+
 def get_version(checkpoint: dict) -> str:
     return checkpoint["pytorch-lightning_version"]
 
@@ -121,9 +124,6 @@ def set_version(checkpoint: dict, version: str):
     checkpoint["pytorch-lightning_version"] = version
 
 
-all_migrations = dict((ver, default_migration) for ver in version_history)
-
-
 class Migration:
     """ Decorator for a function that upgrades a checkpoint from one version to the next. """
 

diff --git a/pytorch_lightning/utilities/migration/migrations.py b/pytorch_lightning/utilities/migration/migrations.py
@@ -4,14 +4,6 @@
 from pytorch_lightning.utilities.migration.patch import pl_legacy_patch
 
 
-@Migration(target="1.2.8")
-def upgrade_callback_names(checkpoint: dict) -> dict:
-    if "callbacks" not in checkpoint:
-        return checkpoint
-    checkpoint["callbacks"] = reversed(checkpoint["callbacks"])
-    return checkpoint
-
-
 @Migration(target="1.2.8")
 def upgrade_something_else(checkpoint: dict) -> dict:
     return checkpoint