Skip to content

Commit

Permalink
Fix mypy error by inheriting from Iterator
Browse files Browse the repository at this point in the history
  • Loading branch information
qwhelan committed May 5, 2024
1 parent 2229514 commit 691dcfa
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/rp2/abstract_entry_set.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

from collections import Iterator
from copy import copy
from datetime import date, datetime
from typing import Dict, List, Optional, Set
Expand Down Expand Up @@ -167,7 +168,7 @@ def __iter__(self) -> "EntrySetIterator":
return EntrySetIterator(self)


class EntrySetIterator:
class EntrySetIterator(Iterator):
def __init__(self, entry_set: AbstractEntrySet) -> None:
self.__entry_set: AbstractEntrySet = entry_set
self.__entry_set_size: int = self.__entry_set.count
Expand Down

0 comments on commit 691dcfa

Please sign in to comment.