tarzanagh commited on
Commit
445f095
·
verified ·
1 Parent(s): 4c67792

Update benchmarks/run.py: remove external references

Browse files
Files changed (1) hide show
  1. benchmarks/run.py +4 -4
benchmarks/run.py CHANGED
@@ -86,10 +86,10 @@ GRAPH_TYPE_DEFAULTS = {
86
 
87
 
88
  def run_passage_entity(args):
89
- """Run benchmark using passage-entity graph (hipporag_pipeline)."""
90
  # Bypass src/__init__.py (heavy AWS deps)
91
  import types as _types
92
- for _pkg_path in ["src", "src.retrievers", "src.hipporag_pipeline"]:
93
  if _pkg_path not in sys.modules:
94
  _m = _types.ModuleType(_pkg_path)
95
  _m.__path__ = [os.path.join(QAFD_RAG_HOME, *_pkg_path.split("."))]
@@ -109,7 +109,7 @@ def run_passage_entity(args):
109
  _load_mod("src.retrievers.flow_diffusion", os.path.join(_src, "retrievers", "flow_diffusion.py"))
110
 
111
  # Import after module setup
112
- from src.hipporag_pipeline.benchmark_runner import main as hippo_main
113
 
114
  # Build sys.argv for the sub-module
115
  sub_argv = [
@@ -140,7 +140,7 @@ def run_passage_entity(args):
140
  old_argv = sys.argv
141
  sys.argv = sub_argv
142
  try:
143
- hippo_main()
144
  finally:
145
  sys.argv = old_argv
146
 
 
86
 
87
 
88
  def run_passage_entity(args):
89
+ """Run benchmark using passage-entity graph (passage_entity)."""
90
  # Bypass src/__init__.py (heavy AWS deps)
91
  import types as _types
92
+ for _pkg_path in ["src", "src.retrievers", "src.passage_entity"]:
93
  if _pkg_path not in sys.modules:
94
  _m = _types.ModuleType(_pkg_path)
95
  _m.__path__ = [os.path.join(QAFD_RAG_HOME, *_pkg_path.split("."))]
 
109
  _load_mod("src.retrievers.flow_diffusion", os.path.join(_src, "retrievers", "flow_diffusion.py"))
110
 
111
  # Import after module setup
112
+ from src.passage_entity.benchmark_runner import main as pe_main
113
 
114
  # Build sys.argv for the sub-module
115
  sub_argv = [
 
140
  old_argv = sys.argv
141
  sys.argv = sub_argv
142
  try:
143
+ pe_main()
144
  finally:
145
  sys.argv = old_argv
146