SnapShooter: Fixed checking for multi_value on plain attributes.

CESNET · Aug 9, 2023 · f57754a · f57754a
1 parent d44bb66
commit f57754a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dp3/snapshots/snapshooter.py b/dp3/snapshots/snapshooter.py
@@ -517,7 +517,7 @@ def get_linked_entity_ids(self, entity_type: str, current_values: dict) -> set[t
             if (entity_type, attr) not in self.model_spec.relations:
                 continue
             attr_spec = self.model_spec.relations[entity_type, attr]
-            if attr_spec.multi_value:
+            if attr_spec.t == AttrType.OBSERVATIONS and attr_spec.multi_value:
                 related_entity_ids.update((attr_spec.relation_to, v["eid"]) for v in val)
             else:
                 related_entity_ids.add((attr_spec.relation_to, val["eid"]))
@@ -530,7 +530,7 @@ def link_loaded_entities(self, loaded_entities: dict):
                 if (entity_type, attr) not in self.model_spec.relations:
                     continue
                 attr_spec = self.model_spec.relations[entity_type, attr]
-                if attr_spec.multi_value:
+                if attr_spec.t == AttrType.OBSERVATIONS and attr_spec.multi_value:
                     entity[attr] = [
                         {
                             **v,