abinazebinoy commited on
Commit
4e07a37
·
1 Parent(s): 8702ccc

fix(BUG-C1): add BenamiInvestigator class wrapper

Browse files
ai/investigators/benami_investigator.py CHANGED
@@ -61,3 +61,15 @@ class BenamiInvestigator:
61
  session=None, driver=None) -> dict:
62
  return investigate(entity_id, entity_name,
63
  session=session, driver=driver)
 
 
 
 
 
 
 
 
 
 
 
 
 
61
  session=None, driver=None) -> dict:
62
  return investigate(entity_id, entity_name,
63
  session=session, driver=driver)
64
+
65
+
66
+ # BUG-C1 FIX: class wrapper to match __init__.py import expectation
67
+ class BenamiInvestigator:
68
+ NAME = NAME
69
+ FOCUS = FOCUS
70
+ WEIGHT = WEIGHT
71
+
72
+ def investigate(self, entity_id: str, entity_name: str,
73
+ session=None, driver=None) -> dict:
74
+ return investigate(entity_id, entity_name,
75
+ session=session, driver=driver)