Check for Custom Dataset

Hi, Thanks your great work!

I would like to test this on my custom dataset, which is a univariate time series (i.e., it has a single feature). I also need the ability to mask specific blocks of data. Could you please confirm if this functionality can be implemented?
