query_procured_balancing_capacity doesn't work with type_marketagreement_type = 'A01' parameter

[beregmiklos]

Dear Developers,

I'm trying to use the EntsoePandasClient.query_procured_balancing_capacity method with the type_marketagreement_type parameter set to 'A01'. I found out that this type of document is limited and and it exceeds this limit even with the smallest time frame possible, which makes the code ending up in an infinite loop. I know that there's a designated decorator for that but I couldn't find out how to use it correctly.

I really hope you could help me fixing this issue.

[fboerman]

hi @beregmiklos , do you have a short code snippit to demonstrate this issue?

[beregmiklos]

Hi @fboerman, you could use this code to recreate the issue:

start_date = pd.Timestamp(2024, 2, 6, tzinfo = tz.gettz('Europe/Budapest'))
end_date = pd.Timestamp(2024, 2, 7, tzinfo = tz.gettz('Europe/Budapest'))
country_code = 'HU'

df_balancing_capacity = client.query_procured_balancing_capacity(country_code, start = start_date, end = end_date, process_type = 'A51', type_marketagreement_type = 'A01')