iter

Author: AlexanderDokuchaev

Diff for: nncf/common/initialization/dataloader.py

@@ -11,7 +11,7 @@
 """Interface for user-defined data usage during the compression algorithm initialization process."""
 from abc import ABC
 from abc import abstractmethod
-from typing import Any
+from typing import Any, Iterator
 
 from nncf.common.utils.api_marker import api
 
@@ -32,7 +32,7 @@ def batch_size(self) -> int:
         """
 
     @abstractmethod
-    def __iter__(self) -> Any:
+    def __iter__(self) -> Iterator[Any]:
         """
         Creates an iterator for the elements of a custom data source.
         The returned iterator implements the Python Iterator protocol.