AttributeError when running predict.py on patients_like_me task

[pailozka]

Hi,
I'm trying to generate the predictions on the data you have provided for patients_like_me task. Unfortunately, code fails on patient_nca_model.py with the following error: AttributeError: 'function' object has no attribute 'dataloader'.
Here is part of code where is fails precisely. It appears that self.test_dataloader is either missing or does not have dataloader argument.

if save:
            run_folder = Path(project_config.PROJECT_DIR) / 'checkpoints' / 'patient_NCA' / self.hparams.hparams['run_name'] / (Path(self.test_dataloader.dataloader.dataset.filepath).stem ) #.replce('/', '_')
            run_folder.mkdir(parents=True, exist_ok=True)
            print('run_folder', run_folder)