Fix sample loader for user ES model (#234)

amundsen-io · Mar 31, 2020 · 2ee56a6 · 2ee56a6
1 parent ae452b7
commit 2ee56a6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/example/sample_data/sample_user.csv b/example/sample_data/sample_user.csv
@@ -1,4 +1,4 @@
-email,first_name,last_name,full_name,github_username,team_name,employee_type,manager_email,slack_id
-[email protected],Roald,Amundsen,"Roald Amundsen",lyft,"Team Amundsen",sailor,"[email protected]",ramundzn
-[email protected],Christopher,Columbus,"Christopher Columbus",ChristopherColumbusFAKE,"Team Amundsen",sailor,"[email protected]",chrisc
-[email protected],Buzz,Aldrin,"Buzz Aldrin",BuzzAldrinFAKE,"Team Amundsen",astronaut,"[email protected]",buzz
+email,first_name,last_name,full_name,github_username,team_name,employee_type,manager_email,slack_id,role_name
+[email protected],Roald,Amundsen,"Roald Amundsen",lyft,"Team Amundsen",sailor,"[email protected]",ramundzn,swe
+[email protected],Christopher,Columbus,"Christopher Columbus",ChristopherColumbusFAKE,"Team Amundsen",sailor,"[email protected]",chrisc,swe
+[email protected],Buzz,Aldrin,"Buzz Aldrin",BuzzAldrinFAKE,"Team Amundsen",astronaut,"[email protected]",buzz,swe
diff --git a/example/scripts/sample_data_loader.py b/example/scripts/sample_data_loader.py
@@ -285,7 +285,7 @@ def create_es_publisher_sample_job(elasticsearch_index_alias='table_search_index
             return user.email as email, user.first_name as first_name, user.last_name as last_name,
             user.full_name as full_name, user.github_username as github_username, user.team_name as team_name,
             user.employee_type as employee_type, manager.email as manager_email, user.slack_id as slack_id,
-            user.is_active as is_active,
+            user.role_name as role_name, user.is_active as is_active,
             REDUCE(sum_r = 0, r in COLLECT(DISTINCT read)| sum_r + r.read_count) AS total_read,
             count(distinct b) as total_own,
             count(distinct c) AS total_follow