Merge b2edbee into eec6f3c

kensho-technologies · Mar 29, 2019 · 6f049f0 · 6f049f0
2 parents eec6f3c + b2edbee
commit 6f049f0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/graphql_compiler/schema_generation/graphql_schema.py b/graphql_compiler/schema_generation/graphql_schema.py
@@ -355,7 +355,6 @@ def get_graphql_schema_from_schema_graph(schema_graph, class_to_field_type_overr
     RootSchemaQuery = GraphQLObjectType('RootSchemaQuery', OrderedDict([
         (name, GraphQLField(value))
         for name, value in sorted(six.iteritems(graphql_types), key=lambda x: x[0])
-        if not isinstance(value, GraphQLUnionType)
     ]))
 
     schema = GraphQLSchema(RootSchemaQuery, directives=DIRECTIVES)

diff --git a/graphql_compiler/tests/test_helpers.py b/graphql_compiler/tests/test_helpers.py
@@ -167,6 +167,8 @@
       FoodOrSpecies: FoodOrSpecies
       Location: Location
       Species: Species
+      Union__BirthEvent__Event__FeedingEvent: Union__BirthEvent__Event__FeedingEvent
+      Union__Food__FoodOrSpecies__Species: Union__Food__FoodOrSpecies__Species
       UniquelyIdentifiable: UniquelyIdentifiable
     }