Skip to content

Commit

Permalink
Fixed S2 year parameter for date range (#176)
Browse files Browse the repository at this point in the history
  • Loading branch information
de-code authored Aug 23, 2023
1 parent 6bfaddc commit 1a43680
Showing 1 changed file with 9 additions and 8 deletions.
17 changes: 9 additions & 8 deletions sciety_labs/providers/semantic_scholar.py
Original file line number Diff line number Diff line change
Expand Up @@ -318,14 +318,15 @@ def iter_unfiltered_search_result_item(
)


def get_years_for_date_range(
def get_year_request_parameter_for_date_range(
from_date: date,
to_date: date
) -> Sequence[int]:
years = [from_date.year]
while years[-1] < to_date.year:
years.append(years[-1] + 1)
return years
) -> str:
from_year = from_date.year
to_year = to_date.year
if to_year == from_year:
return str(from_year)
return f'{from_year}-{to_year}'


def iter_search_results_published_within_date_range(
Expand Down Expand Up @@ -363,10 +364,10 @@ def iter_search_result_item(
query=search_parameters.query,
additional_search_parameters={
**SEMANTIC_SCHOLAR_SEARCH_PARAMETERS_WITH_VENUES,
'year': ','.join([str(year) for year in get_years_for_date_range(
'year': get_year_request_parameter_for_date_range(
from_date=from_date,
to_date=to_date
)])
)
}
)
if search_parameters.is_evaluated_only:
Expand Down

0 comments on commit 1a43680

Please sign in to comment.