From f958adbff134c91d270f2a689c4981b41457d64c Mon Sep 17 00:00:00 2001
From: protolambda <proto@protolambda.com>
Date: Thu, 11 Apr 2019 19:25:00 +1000
Subject: [PATCH] generator running fixes

---
 Makefile                    | 3 ++-
 test_generators/bls/main.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/Makefile b/Makefile
index bf23a1442..7ca2cc909 100644
--- a/Makefile
+++ b/Makefile
@@ -5,6 +5,7 @@ PY_SPEC_DIR = $(TEST_LIBS_DIR)/pyspec
 PY_TEST_DIR = ./py_tests
 YAML_TEST_DIR = ./yaml_tests
 GENERATOR_DIR = ./test_generators
+CONFIGS_DIR = ./configs
 
 # Collect a list of generator names
 GENERATORS = $(sort $(dir $(wildcard $(GENERATOR_DIR)/*/)))
@@ -57,7 +58,7 @@ define build_yaml_tests
 	# 3) Install all the necessary requirements
 	# 4) Run the generator. The generator is assumed to have an "main.py" file.
 	# 5) We output to the tests dir (generator program should accept a "-o <filepath>" argument.
-	cd $(GENERATOR_DIR)$(1); python3 -m venv venv; . venv/bin/activate; pip3 install -r requirements.txt; python3 main.py -o $(CURRENT_DIR)/$(YAML_TEST_DIR)$(1)
+	cd $(GENERATOR_DIR)$(1); python3 -m venv venv; . venv/bin/activate; pip3 install -r requirements.txt; python3 main.py -o $(CURRENT_DIR)/$(YAML_TEST_DIR)$(1) -c $(CURRENT_DIR)/$(CONFIGS_DIR)
 
 	$(info generator $(1) finished)
 endef
diff --git a/test_generators/bls/main.py b/test_generators/bls/main.py
index e5288c045..1a0373068 100644
--- a/test_generators/bls/main.py
+++ b/test_generators/bls/main.py
@@ -227,7 +227,7 @@ def bls_aggregate_pubkeys_suite(configs_path: str) -> gen_typing.TestSuiteOutput
 
 
 if __name__ == "__main__":
-    gen_runner.run_generator("shuffling", [
+    gen_runner.run_generator("bls", [
         bls_msg_hash_compressed_suite,
         bls_msg_hash_uncompressed_suite,
         bls_priv_to_pub_suite,