add main block for benchmark script

deeppavlov · Oct 18, 2024 · ab266ea · ab266ea
1 parent 96ebd44
commit ab266ea
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/utils/db_benchmark/benchmark_dbs.py b/utils/db_benchmark/benchmark_dbs.py
@@ -36,11 +36,12 @@
 }
 
 
-for db_name, db_uri in dbs.items():
-    benchmark_all(
-        benchmark_dir / f"{db_name}.json",
-        db_name,
-        description="Basic configs",
-        db_uri=db_uri,
-        benchmark_configs=basic_configurations,
-    )
+if __name__ == "__main__":
+    for db_name, db_uri in dbs.items():
+        benchmark_all(
+            benchmark_dir / f"{db_name}.json",
+            db_name,
+            description="Basic configs",
+            db_uri=db_uri,
+            benchmark_configs=basic_configurations,
+        )