fix: fix a wrongly refactored code from previous commit

blb-ventures · Jun 23, 2023 · fb0de57 · fb0de57
1 parent 493a1ad
commit fb0de57
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/strawberry_django_plus/__init__.py b/strawberry_django_plus/__init__.py
@@ -36,7 +36,7 @@ def _process_type(cls, *args, **kwargs):
     type_def = get_object_definition(ret, strict=True)
     for d in type_def.directives or []:
         if isinstance(d, SchemaDirectiveWithResolver):
-            d.register(ret)
+            d.register(type_def)
 
     return ret
 

diff --git a/strawberry_django_plus/optimizer.py b/strawberry_django_plus/optimizer.py
@@ -71,8 +71,7 @@
 _interfaces: """
 defaultdict[
     Schema,
-    Dict[StrawberryObjectDefinition,
-    List[StrawberryObjectDefinition]],
+    Dict[StrawberryObjectDefinition, List[StrawberryObjectDefinition]],
 ]""" = defaultdict(
     dict,
 )