Add to init

frazane · Jun 18, 2024 · c1baf92 · c1baf92
1 parent 34828f5
commit c1baf92
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/scoringrules/core/energy/__init__.py b/scoringrules/core/energy/__init__.py
@@ -1,17 +1,25 @@
 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
 
 __all__ = [
     "nrg",
     "ownrg",
     "vrnrg",
+    "iidnrg",
+    "kbnrg",
     "_energy_score_gufunc",
     "_owenergy_score_gufunc",
     "_vrenergy_score_gufunc",
+    "_energy_score_kband_gufunc",
+    "_energy_score_iid_gufunc",
 ]