diff --git a/scoringrules/core/energy/__init__.py b/scoringrules/core/energy/__init__.py index af64592..f54273f 100644 --- a/scoringrules/core/energy/__init__.py +++ b/scoringrules/core/energy/__init__.py @@ -1,9 +1,13 @@ from ._gufuncs import ( _energy_score_gufunc, + _energy_score_iid_gufunc, + _energy_score_kband_gufunc, _owenergy_score_gufunc, _vrenergy_score_gufunc, ) from ._score import energy_score as nrg +from ._score import energy_score_iid as iidnrg +from ._score import energy_score_kband as kbnrg from ._score import owenergy_score as ownrg from ._score import vrenergy_score as vrnrg @@ -11,7 +15,11 @@ "nrg", "ownrg", "vrnrg", + "iidnrg", + "kbnrg", "_energy_score_gufunc", "_owenergy_score_gufunc", "_vrenergy_score_gufunc", + "_energy_score_kband_gufunc", + "_energy_score_iid_gufunc", ]