Skip to content

Commit a35175c

Browse files
committed
fix uninitialised *_tmp fields in QMRATreatment
1 parent c170ece commit a35175c

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

qmra/risk_assessment/qmra_models.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -311,9 +311,12 @@ def save(
311311
self, force_insert=False, force_update=False, using=None, update_fields=None
312312
):
313313
super().save(force_insert, force_update, using, update_fields)
314-
self.bacteria_references.set(self._bacteria_references_tmp)
315-
self.viruses_references.set(self._viruses_references_tmp)
316-
self.protozoa_references.set(self._protozoa_references_tmp)
314+
if hasattr(self, "_viruses_references_tmp"):
315+
self.viruses_references.set(self._viruses_references_tmp)
316+
if hasattr(self, "_protozoa_references_tmp"):
317+
self.protozoa_references.set(self._protozoa_references_tmp)
318+
if hasattr(self, "_bacteria_references_tmp"):
319+
self.bacteria_references.set(self._bacteria_references_tmp)
317320
return self
318321

319322

0 commit comments

Comments
 (0)