fix: : could not broadcast input array from shape (2,) into shape (1,) (

#225)

pyproject.toml

@@ -29,6 +29,7 @@ swifter = "^1.4.0"
 tenacity = "^8.2.3"
 sentry-sdk = "^2.13.0"
 modin = {extras = ["ray"], version = "^0.31.0"}
+numpy = "1.26.4"
 [build-system]
 requires = ["poetry-core>=1.1"]
 build-backend = "poetry.core.masonry.api"

quotaclimat/data_processing/mediatree/api_import.py

@@ -122,7 +122,8 @@ async def get_and_save_api_data(exit_event):
             exit_event.set()
         except Exception as err:
             logging.fatal("get_and_save_api_data (%s) %s" % (type(err).__name__, err))
-            sys.exit(0)
+            ray.shutdown()
+            sys.exit(1)
 
 # "Randomly wait up to 2^x * 1 seconds between each retry until the range reaches 60 seconds, then randomly up to 60 seconds afterwards"
 # @see https://github.com/jd/tenacity/tree/main
@@ -294,7 +295,8 @@ async def main():
             res=health_check_task.cancel()
         except Exception as err:
             logging.fatal("Main crash (%s) %s" % (type(err).__name__, err))
-            sys.exit(0)
+            ray.shutdown()
+            sys.exit(1)
     logging.info("Exiting with success")
     sys.exit(0)
 

quotaclimat/data_processing/mediatree/channel_program.py

@@ -88,10 +88,12 @@ def get_matching_program_weekday(df_program: pd.DataFrame, start_time: pd.Timest
     df_program["weekday_mask"] = df_program['weekday'].apply(
         lambda x: compare_weekday(x, start_weekday)
     )
+    logging.debug("weekday_mask done")
     matching_rows =  df_program[
                         (df_program['channel_name'] == channel_name) &
                           df_program["weekday_mask"] == True
                         ]
+    logging.debug("matching_rows done")
     matching_rows.drop(columns=['weekday_mask'], inplace=True)
     matching_rows.drop(columns=['weekday'], inplace=True)
 

test/sitemap/test_program_metadata.py

@@ -199,6 +199,13 @@ def test_get_13h_program_with_start_timestamp():
     assert program_name == "13h15 le samedi"
     assert program_type == "Information - Journal"
 
+def test_get_13h_monday_program_with_start_timestamp():
+    df_programs = get_programs()
+    monday_13h18 = 1724671082
+    program_name, program_type = get_a_program_with_start_timestamp(df_programs, pd.to_datetime(monday_13h18, unit='s', utc=True).tz_convert('Europe/Paris'), channel_name)
+    assert program_name == "JT 13h"
+    assert program_type == "Information - Journal"
+
 def test_compare_weekday_string():
     assert compare_weekday('*', 0) == True
     assert compare_weekday('*', 3) == True