Skip to content

Commit

Permalink
Simplify init for EMNIST
Browse files Browse the repository at this point in the history
  • Loading branch information
pczarnik committed Jul 6, 2024
1 parent 589714b commit 144b5bd
Showing 1 changed file with 2 additions and 9 deletions.
11 changes: 2 additions & 9 deletions mnists/_emnist.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
from typing import Optional

from .dataset import SplitDataset, ZippedDataset


Expand Down Expand Up @@ -41,13 +39,8 @@ class EMNIST(SplitDataset):

resources = {"gzip": ("gzip.zip", "58c8d27c78d21e728a6bc7b3cc06412e")}

def __init__(
self,
target_dir: Optional[str] = None,
download: bool = True,
force_download: bool = False,
) -> None:
super().__init__(target_dir, download, force_download)
def __init__(self, *args, **kwargs) -> None:
super().__init__(*args, **kwargs)

self.Balanced = self._create_split(Balanced)
self.ByClass = self._create_split(ByClass)
Expand Down

0 comments on commit 144b5bd

Please sign in to comment.