Fix bug in check for collate groups key (#907)

dssg · Aug 9, 2022 · f0a74a5 · f0a74a5
1 parent 3ae02af
commit f0a74a5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/triage/experiments/validate.py b/src/triage/experiments/validate.py
@@ -19,9 +19,10 @@
 
 
 class Validator:
-    def __init__(self, db_engine=None, strict=True):
+    def __init__(self, db_engine=None, strict=True, entity_id_column="entity_id"):
         self.db_engine = db_engine
         self.strict = strict
+        self.entity_id_column = entity_id_column
 
     def run(self, *args, **kwargs):
         try: