Update __init__.py

EliHei2 · Dec 11, 2024 · 5b84d3e · 5b84d3e
1 parent f6b9ccd
commit 5b84d3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/segger/prediction/__init__.py b/src/segger/prediction/__init__.py
@@ -4,6 +4,6 @@
 Contains prediction scripts and utilities for the Segger model.
 """
 
-__all__ = ["load_model", "predict"]
+__all__ = ["load_model", "segment"]
 
-from .predict import load_model, predict_parquet
+from .predict_parquet import load_model, segmenet