remihndz

As a simpler workaround, adding the lines: ```python if 'primus' in str(trainer_class).lower(): trainer_class = trainer_class(plans=plans, configuration=configuration_name, fold=None, dataset_json=dataset_json, device=self.device) ``` in `inference/predict_from_raw_data.py`, line 104 seems to work.