cleanup

src/accelerate/data_loader.py

@@ -435,9 +435,8 @@ def load_state_dict(self, state_dict):
     @property
     def __class__(self):
         """
-        In order to maintain backwards compatability with other code, we need to ensure `isinstance(obj, DataLoader)`
-        returs true. This is because some downstream code assumes that the `DataLoader` is the base class of the
-        object.
+        In order to maintain backwards compatability with other code, we need to ensure `isinstance(obj, DataLoader)` returs true.
+        This is because some downstream code assumes that the `DataLoader` is the base class of the object.
         """
         return self.base_dataloader.__class__
 
@@ -480,18 +479,6 @@ def _update_state_dict(self):
             # Then tag if we are at the end of the dataloader
             self.dl_state_dict["_iterator_finished"] = self.end_of_dataloader
 
-
-class DataLoaderAdapterImpl(DataLoaderAdapter, DataLoader):
-    pass
-
-
-if is_torchdata_stateful_dataloader_available():
-    from torchdata.stateful_dataloader import StatefulDataLoader
-
-    class StatefulDataLoaderAdapterImpl(DataLoaderAdapter, StatefulDataLoader):
-        pass
-
-
 class DataLoaderShard(DataLoaderAdapter, DataLoaderStateMixin):
     """
     Subclass of `DataLoaderAdapter` that will deal with device placement and current distributed setup.