instead of using direct methods using enter and exit.

[hariharan-devarajan]

Instead of

        global profile_args
        profile_args["epoch"] = epoch_number
        t0 = time()
        super().__init__(format_type, dataset_type, epoch_number)
        self._dataset = None
        t1 = time()
        PerfTrace.get_instance().event_complete(f"{self.__init__.__qualname__}", MY_MODULE, t0, t1 - t0,
                                                arguments=profile_args)

We do just this

        with Profile(name=f"{self.__init__.__qualname__}", cat=MODULE_DATA_LOADER, epoch=epoch_number):
            super().__init__(format_type, dataset_type, epoch_number)
            self._dataset = None