diff --git a/git4intel/client.py b/git4intel/client.py index 00a0321..e13ee49 100644 --- a/git4intel/client.py +++ b/git4intel/client.py @@ -108,8 +108,6 @@ def store_core_data(self): def register_ident(self, id_bundle, _type): # Must only contain a id obj and a location ref # _type must be the relevant class (org or individual) - if len(id_bundle['objects']) != 2: - return False for obj in id_bundle['objects']: obj_type = str(obj['id']).split('--')[0] @@ -347,7 +345,9 @@ def get_objects(self, user_id, obj_ids, values=None): if values: if not isinstance(values, list): return False - q = {"query": {"bool": {"must": []}}} + q = { + "query": {"bool": {"must": []}}, + } id_q = {"bool": {"should": []}} for obj_id in obj_ids: id_q["bool"]["should"].append(