minor clean up

merlin/models/loader/backend.py

@@ -31,8 +31,6 @@ def _get_dataset_schema(dataset):
     return dataset.schema if hasattr(dataset, "schema") else None
 
 
-# TODO: implement as metaclass and assign methods to children
-# to avoid having to do Dataset.<method> calls?
 class DataLoader(LoaderBase):
     def __init__(
         self,
@@ -94,11 +92,6 @@ def __init__(
                 "label_names properties or supply a schema.pbtxt file in dataset directory."
             )
 
-        self.__buff = None
-        self.__buff_len = None
-        self._batch_itr = None
-        self._workers = None
-
     @annotate("make_tensors", color="darkgreen", domain="merlin_dataloader")
     def make_tensors(self, gdf, use_nnz=False):
         split_idx = self._get_segment_lengths(len(gdf))

merlin/models/tf/loader.py

@@ -18,6 +18,7 @@
 from typing import Protocol
 
 import dask.dataframe as dd
+import numpy as np
 import tensorflow as tf
 
 import merlin.loader.tensorflow
@@ -285,6 +286,7 @@ def __init__(
             global_size = global_size or hvd.size()
             global_rank = global_rank or hvd.rank()
             seed_fn = seed_fn or get_default_hvd_seed_fn()
+
         DataLoader.__init__(
             self,
             dataset,