Bug fix in get_queryset while filtering generic lazy reference field

Author: arunsureshkumar

graphene_mongo/fields.py

@@ -213,8 +213,12 @@ def get_queryset(self, model, info, required_fields=list(), skip=None, limit=Non
                     hydrated_references[arg_name] = reference_obj
                 elif arg_name in self.model._fields_ordered and isinstance(getattr(self.model, arg_name),
                                                                            mongoengine.fields.GenericReferenceField):
-                    reference_obj = get_document(self.registry._registry_string_map[from_global_id(arg)[0]])(
-                        pk=from_global_id(arg)[1])
+                    try:
+                        reference_obj = get_document(self.registry._registry_string_map[from_global_id(arg)[0]])(
+                            pk=from_global_id(arg)[1])
+                    except TypeError:
+                        reference_obj = get_document(arg["_cls"])(
+                            pk=arg["_ref"].id)
                     hydrated_references[arg_name] = reference_obj
                 elif arg_name == "id":
                     hydrated_references["id"] = from_global_id(args.pop("id", None))[1]