diff --git a/hub/management/commands/import_gbgw_events.py b/hub/management/commands/import_gbgw_events.py index 6d0235bcf..69f697d9b 100644 --- a/hub/management/commands/import_gbgw_events.py +++ b/hub/management/commands/import_gbgw_events.py @@ -54,6 +54,7 @@ def add_data_sets(self, df): ) self.data_type = data_type + self.data_types[data_type.name] = data_type def process_data(self, df): AreaData.objects.filter(data_type=self.data_type).delete() diff --git a/hub/management/commands/import_gbgw_events_23.py b/hub/management/commands/import_gbgw_events_23.py index 8bb23d490..9dd40fe52 100644 --- a/hub/management/commands/import_gbgw_events_23.py +++ b/hub/management/commands/import_gbgw_events_23.py @@ -54,6 +54,7 @@ def add_data_sets(self, df): ) self.data_type = data_type + self.data_types[data_type.name] = data_type def process_data(self, df): AreaData.objects.filter(data_type=self.data_type).delete()