abinazebinoy commited on
Commit
805478a
·
1 Parent(s): 529f177

fix(BUG-M9): pass driver kwarg to investigators in run_investigator -- BenamiDetector/SpectralAnalyzer need driver not session

Browse files
Files changed (1) hide show
  1. ai/multi_investigator.py +5 -1
ai/multi_investigator.py CHANGED
@@ -20,7 +20,11 @@ def run_investigator(investigator_class, entity_id: str,
20
  try:
21
  if driver:
22
  with driver.session() as session:
23
- return investigator.investigate(entity_id, entity_name, session)
 
 
 
 
24
  else:
25
  # BUG-11 FIX: warn explicitly when FakeSession is used -- previously
26
  # the report looked complete but all findings were empty with no
 
20
  try:
21
  if driver:
22
  with driver.session() as session:
23
+ # BUG-M9 FIX: pass driver as kwarg for module-style investigators
24
+ # (BenamiDetector and SpectralAnalyzer need driver, not session)
25
+ return investigator.investigate(
26
+ entity_id, entity_name, session=session, driver=driver
27
+ )
28
  else:
29
  # BUG-11 FIX: warn explicitly when FakeSession is used -- previously
30
  # the report looked complete but all findings were empty with no