fix: fix return type of load function

tinkoff-ai · May 23, 2023 · 2c5a491 · 2c5a491
1 parent 2073fe9
commit 2c5a491
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/etna/core/utils.py b/etna/core/utils.py
@@ -9,8 +9,10 @@
 
 from hydra_slayer import get_factory
 
+from etna.core.saving import AbstractSaveable
 
-def load(path: pathlib.Path, **kwargs: Any) -> Any:
+
+def load(path: pathlib.Path, **kwargs: Any) -> AbstractSaveable:
     """Load saved object by path.
 
     Parameters