71a7b91abfd37da9a024e4106fbde8646b524ad7,scripts/radio_util.py,Generator,_read_chunck,#Generator#,33

Before Change

        print("Number of files used: %s"%len(files))
    
    def _read_chunck(self):
        idx = self.idxs[self.iter]
        with h5py.File(self.files[self.file_idx], "r") as fp:
            sl = slice((idx*self.nx), ((idx+1)*self.nx))
            data = fp["data"][:, sl]
            rfi = fp["mask"][:, sl]

After Change

    def _read_chunck(self):
        with h5py.File(self.files[self.file_idx], "r") as fp:
            nx = fp["data"].shape[1]
            idx = np.random.choice(np.arange(nx/self.nx).astype(np.int))
            
            sl = slice((idx*self.nx), ((idx+1)*self.nx))
            data = fp["data"][:, sl]
            rfi = fp["mask"][:, sl]

Instances