hefty 0.1.2__tar.gz → 0.1.4__tar.gz

{hefty-0.1.2/src/hefty.egg-info → hefty-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hefty
-Version: 0.1.2
+Version: 0.1.4
 Summary: Some (relatively) lightweight short-term energy forecasting tools for solar, wind, and load.
 Author: Will Hobbs
 License-Expression: BSD-3-Clause

{hefty-0.1.2 → hefty-0.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "hefty"
-version = "0.1.2"
+version = "0.1.4"
 description = "Some (relatively) lightweight short-term energy forecasting tools for solar, wind, and load."
 readme = "README.md"
 requires-python = ">=3.11"

{hefty-0.1.2 → hefty-0.1.4}/src/hefty/solar.py

@@ -1249,6 +1249,11 @@ def get_solar_forecast_ensemble_subset(
 
     # get deterministic temp_air
     search_str = ':2t:sfc:g:0001:od:cf:enfo'
+    product = 'enfo'
+    # account for IFS changes starting 2026-05-12 06z
+    if (init_date.tz_localize(None) > pd.to_datetime('2026-05-12')):
+        product = 'oper'
+        search_str = '2t:sfc'
 
     # try n times based loosely on
     # https://thingspython.wordpress.com/2021/12/05/how-to-try-something-n-times-in-python/
@@ -1259,7 +1264,7 @@ def get_solar_forecast_ensemble_subset(
                 # try downloading
                 ds = FastHerbie(DATES=[init_date],
                                 model=model_herbie,
-                                product='enfo',
+                                product=product,
                                 fxx=fxx_range,
                                 priority=priority).xarray(search_str)
             else:
@@ -1267,7 +1272,7 @@ def get_solar_forecast_ensemble_subset(
                 # partial files
                 ds = FastHerbie(DATES=[init_date],
                                 model=model_herbie,
-                                product='enfo',
+                                product=product,
                                 fxx=fxx_range,
                                 priority=priority).xarray(search_str,
                                                           overwrite=True)
@@ -1687,10 +1692,16 @@ def get_solar_forecast_ensemble(latitude, longitude, init_date, run_length,
             # get deterministic temp_air using ifs control member
             search_str = ':2t:sfc:g:0001:od:cf:enfo'
             get_control = None
+            product = 'enfo'
+            # account for IFS changes starting 2026-05-12 06z
+            if (init_date.tz_localize(None) > pd.to_datetime('2026-05-12')):
+                product = 'oper'
+                search_str = '2t:sfc'
         elif model == 'aifs_ens':
             search_str = ':2t:sfc:'
             # Herbie kwarg to get control member, https://herbie.readthedocs.io/en/stable/gallery/ecmwf_models/ecmwf.html#AIFS-Ensembles
             get_control = True
+            product = 'enfo'
 
         # try n times based loosely on
         # https://thingspython.wordpress.com/2021/12/05/how-to-try-something-n-times-in-python/
@@ -1701,7 +1712,7 @@ def get_solar_forecast_ensemble(latitude, longitude, init_date, run_length,
                     # try downloading
                     FH = FastHerbie(DATES=[init_date],
                                     model=model_herbie,
-                                    product='enfo',
+                                    product=product,
                                     fxx=fxx_range,
                                     priority=priority,
                                     get_control=get_control,
@@ -1713,7 +1724,7 @@ def get_solar_forecast_ensemble(latitude, longitude, init_date, run_length,
                     # partial files
                     FH = FastHerbie(DATES=[init_date],
                                     model=model_herbie,
-                                    product='enfo',
+                                    product=product,
                                     fxx=fxx_range,
                                     priority=priority,
                                     get_control=get_control,

{hefty-0.1.2 → hefty-0.1.4}/src/hefty/utilities.py

@@ -131,7 +131,7 @@ def get_fcast_definition(model='gfs'):
                        '2024-03-10 12:00',
                        '2024-03-10 18:00'],  # https://herbie.readthedocs.io/en/stable/gallery/ecmwf_models/ecmwf.html#Data-Availability
         'start_hour': [0, 150, 0],
-        'end_hour': [144, 240, 144],
+        'end_hour': [144, 360, 144],
         'interval': [3, 6, 3],
         'first_cycle': [0, 0, 6],
         'update_period': [12, 12, 12],
@@ -760,7 +760,13 @@ def model_input_formatter(init_date, run_length, lead_time_to_start=0,
             fxx_range = range(lead_time_to_start, fxx_max + 1, 3)
 
         # Herbie inputs
-        if init_date.hour == 6 or init_date.hour == 18:
+        init_date = pd.to_datetime(init_date)
+        # scda goes away/went away 2026-05-12
+        # see https://confluence.ecmwf.int/display/FCST/Implementation+of+IFS+Cycle+50r1
+        if (init_date.tz_localize(None) <
+            pd.to_datetime('2026-05-12 06:00') and
+            (init_date.hour == 6 or
+             init_date.hour == 18)):
             product = 'scda'
         else:
             product = 'oper'

{hefty-0.1.2 → hefty-0.1.4/src/hefty.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hefty
-Version: 0.1.2
+Version: 0.1.4
 Summary: Some (relatively) lightweight short-term energy forecasting tools for solar, wind, and load.
 Author: Will Hobbs
 License-Expression: BSD-3-Clause