    def create_data_loader(self, df):
        """
        Generic data loader function

        :param df: Input tensor

        :return: Returns the constructed dataloader
        """
        return DataLoader(
            df, batch_size=self.args["batch_size"], num_workers=self.args["num_workers"]
        )
Generic data loader function

:param df: Input tensor

:return: Returns the constructed dataloader
tests/integration/frameworks/mlflow/MNIST/mnist_autolog_example.py