Skip to content

Commit 6d38e58

Browse files
authored
Merge pull request #96 from nasaharvest/bugfix-default-labels
Fix end date in the exporter's load_default_labels function
2 parents 549181d + dadd91a commit 6d38e58

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

cropharvest/eo/eo.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -138,13 +138,15 @@ def load_default_labels(
138138
) -> geopandas.GeoDataFrame:
139139
labels = geopandas.read_file(DATAFOLDER_PATH / LABELS_FILENAME)
140140
export_end_year = pd.to_datetime(labels[RequiredColumns.EXPORT_END_DATE]).dt.year
141-
labels["end_date"] = export_end_year.apply(lambda x: date(x, 12, 12))
141+
labels["end_date"] = export_end_year.apply(
142+
lambda x: date(x, EXPORT_END_MONTH, EXPORT_END_DAY)
143+
)
142144
labels = labels.assign(
143145
start_date=lambda x: x["end_date"]
144146
- timedelta(days=DAYS_PER_TIMESTEP * DEFAULT_NUM_TIMESTEPS)
145147
)
146-
labels = labels.assign(
147-
export_identifier=lambda x: f"{x['index']}-{x[RequiredColumns.DATASET]}"
148+
labels["export_identifier"] = labels.apply(
149+
lambda x: f"{x['index']}-{x[RequiredColumns.DATASET]}", axis=1
148150
)
149151
if dataset:
150152
labels = labels[labels.dataset == dataset]

0 commit comments

Comments
 (0)