diff --git a/rosie/core/__init__.py b/rosie/core/__init__.py index 078c55c..a75f30e 100644 --- a/rosie/core/__init__.py +++ b/rosie/core/__init__.py @@ -28,7 +28,7 @@ def __init__(self, settings, adapter): self.settings = settings self.dataset = adapter.dataset self.data_path = adapter.path - if (UNIQUE_IDS): + if self.settings.UNIQUE_IDS: self.suspicions = self.dataset[self.settings.UNIQUE_IDS].copy() else: self.suspicions = self.dataset.copy()