Apply suggestions from code review

Co-authored-by: Philip Meier <[email protected]>

Author: lezwon

torchvision/prototype/datasets/_builtin/usps.py

@@ -53,7 +53,7 @@ def _make_info(self) -> DatasetInfo:
     def resources(self, config: DatasetConfig) -> List[OnlineResource]:
         return [USPS._RESOURCES[config.split]]
 
-    def _prepare_sample(self, data: Tuple[torch.Tensor, torch.Tensor], *, config: DatasetConfig) -> Dict[str, Any]:
+    def _prepare_sample(self, data: Tuple[torch.Tensor, torch.Tensor]) -> Dict[str, Any]:
         image, label = data
         return dict(
             image=Image(image),
@@ -69,4 +69,4 @@ def _make_datapipe(
         dp = USPSFileReader(resource_dps[0])
         dp = hint_sharding(dp)
         dp = hint_shuffling(dp)
-        return Mapper(dp, functools.partial(self._prepare_sample, config=config))
+        return Mapper(dp, self._prepare_sample)