Reformat POL to standardized format

[osm/POL.git] / osm_policy_module / migrations / 004_add_fields_to_alarm.py

diff --git a/osm_policy_module/migrations/004_add_fields_to_alarm.py b/osm_policy_module/migrations/004_add_fields_to_alarm.py
index 44c495a..f88a329 100644 (file)
--- a/osm_policy_module/migrations/004_add_fields_to_alarm.py
+++ b/osm_policy_module/migrations/004_add_fields_to_alarm.py
@@ -52,11 +52,13 @@ SQL = pw.SQL
 def migrate(migrator, database, fake=False, **kwargs):
     """Write your migrations here."""
 
-    migrator.add_fields('scalingalarm',
-                        last_status=pw.CharField(max_length=255, default='insufficient-data'))
+    migrator.add_fields(
+        "scalingalarm",
+        last_status=pw.CharField(max_length=255, default="insufficient-data"),
+    )
 
 
 def rollback(migrator, database, fake=False, **kwargs):
     """Write your rollback migrations here."""
 
-    migrator.remove_fields('scalingalarm', 'last_status')
+    migrator.remove_fields("scalingalarm", "last_status")