Add missing config (#1466)

Signed-off-by: Orion Delwaterman <delwaterman@gmail.com>
amundsen-io · Aug 31, 2021 · 6ae2548 · 6ae2548
1 parent 4a023d6
commit 6ae2548
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/databuilder/example/scripts/sample_postgres_loader.py b/databuilder/example/scripts/sample_postgres_loader.py
@@ -121,6 +121,7 @@ def create_es_publisher_sample_job(elasticsearch_index_alias='table_search_index
     elasticsearch_new_index_key = 'tables' + str(uuid.uuid4())
 
     job_config = ConfigFactory.from_dict({
+        'extractor.search_data.entity_type': 'table',
         f'extractor.search_data.extractor.neo4j.{Neo4jExtractor.GRAPH_URL_CONFIG_KEY}': neo4j_endpoint,
         f'extractor.search_data.extractor.neo4j.{Neo4jExtractor.MODEL_CLASS_CONFIG_KEY}': model_name,
         f'extractor.search_data.extractor.neo4j.{Neo4jExtractor.NEO4J_AUTH_USER}': neo4j_user,