Skip to content

Commit

Permalink
fixup! import_mixin -> actions
Browse files Browse the repository at this point in the history
  • Loading branch information
bouttier committed Sep 13, 2024
1 parent 6a258b0 commit 5b73ada
Showing 1 changed file with 7 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -133,7 +133,7 @@ def check_habitat_dataframe(imprt):

### Dataframe checks
df = load_transient_data_in_dataframe(imprt, entity_habitat, source_cols)
updated_cols |= OcchabImportMixin.dataframe_checks(
updated_cols |= OcchabImportActions.dataframe_checks(
imprt, df, entity_habitat, fields, selected_fields
)
update_transient_data_from_dataframe(imprt, entity_habitat, updated_cols, df)
Expand Down Expand Up @@ -267,7 +267,7 @@ def check_station_dataframe(imprt):
### Dataframe checks
df = load_transient_data_in_dataframe(imprt, entity_station, source_cols)

updated_cols |= OcchabImportMixin.dataframe_checks(
updated_cols |= OcchabImportActions.dataframe_checks(
imprt, df, entity_station, fields, selected_fields
)
updated_cols |= check_datasets(
Expand Down Expand Up @@ -415,15 +415,15 @@ def check_transient_data(task, logger, imprt: TImports):

# We first check station consistency in order to avoid checking
# incoherent station data
OcchabImportMixin.check_station_consistency(imprt)
OcchabImportActions.check_station_consistency(imprt)

# We run station & habitat dataframe checks before SQL checks in order to avoid
# check_types overriding generated values during SQL checks.
OcchabImportMixin.check_station_dataframe(imprt)
OcchabImportMixin.check_habitat_dataframe(imprt)
OcchabImportActions.check_station_dataframe(imprt)
OcchabImportActions.check_habitat_dataframe(imprt)

OcchabImportMixin.check_station_sql(imprt)
OcchabImportMixin.check_habitat_sql(imprt)
OcchabImportActions.check_station_sql(imprt)
OcchabImportActions.check_habitat_sql(imprt)

task.update_state(state="PROGRESS", meta={"progress": 1})

Expand Down

0 comments on commit 5b73ada

Please sign in to comment.