Mypy fixes

SEED-platform · Feb 16, 2024 · c290530 · c290530
1 parent 4b30176
commit c290530
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/seed/tests/test_property_views.py b/seed/tests/test_property_views.py
@@ -38,7 +38,6 @@
     Column,
     ColumnMappingProfile,
     InventoryDocument,
-    Label,
     Meter,
     MeterReading,
     Note,
@@ -47,6 +46,7 @@
     PropertyState,
     PropertyView,
     Scenario,
+    StatusLabel,
     TaxLotProperty,
     TaxLotView
 )
@@ -756,7 +756,7 @@ def setUp(self):
         self.view = self.property_view_factory.get_property_view(cycle=self.cycle)
         self.taxlot_view = self.taxlot_view_factory.get_taxlot_view(cycle=self.cycle)
         self.property = Property.objects.create(organization=self.org, access_level_instance=self.org.root)
-        self.label = Label.objects.create(color="red", name="test_label", super_organization=self.org,)
+        self.label = StatusLabel.objects.create(color="red", name="test_label", super_organization=self.org,)
         self.view.labels.add(self.label)
         self.view.property = self.property
         self.view.save()

diff --git a/seed/tests/test_taxlot_views.py b/seed/tests/test_taxlot_views.py
@@ -16,9 +16,9 @@
     DATA_STATE_MAPPING,
     VIEW_LIST_TAXLOT,
     Column,
-    Label,
     Note,
     PropertyView,
+    StatusLabel,
     TaxLot,
     TaxLotProperty,
     TaxLotView
@@ -406,7 +406,7 @@ def setUp(self):
         self.cycle = self.cycle_factory.get_cycle()
         self.view = self.taxlot_view_factory.get_taxlot_view(cycle=self.cycle)
         self.taxlot = TaxLot.objects.create(organization=self.org, access_level_instance=self.org.root)
-        self.label = Label.objects.create(color="red", name="test_label", super_organization=self.org,)
+        self.label = StatusLabel.objects.create(color="red", name="test_label", super_organization=self.org,)
         self.view.labels.add(self.label)
         self.view.taxlot = self.taxlot
         self.view.save()