diff --git a/elbevalidate/__init__.py b/elbevalidate/__init__.py index d1a26b277..84c36dd81 100644 --- a/elbevalidate/__init__.py +++ b/elbevalidate/__init__.py @@ -235,7 +235,7 @@ def __init__(self, tar): @classmethod @contextlib.contextmanager - def from_file(cls, path) -> collections.abc.Generator['Image', None, None]: + def from_file(cls, path) -> collections.abc.Generator['Tar', None, None]: """ Construct an :py:class:`Tarball` from a local file. """ with tarfile.open(path) as tar: yield cls(tar)