Update benchmarks/run.py: remove external references
Browse files- 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 (
|
| 90 |
# Bypass src/__init__.py (heavy AWS deps)
|
| 91 |
import types as _types
|
| 92 |
-
for _pkg_path in ["src", "src.retrievers", "src.
|
| 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.
|
| 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 |
-
|
| 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 |
|