Shorten kwargs setup

cb109 · Dec 9, 2020 · cef7415 · cef7415
1 parent 8723a96
commit cef7415
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/model_bakery/baker.py b/model_bakery/baker.py
@@ -541,11 +541,10 @@ def generate_value(self, field: Field, commit: bool = True) -> Any:
                 # Pass instance instead of pk/<to_field> value. In case a
                 # custom 'to_field' is set, Django should populate 
                 # field.to_fields with a single string.
-                key = "pk"
+                default_key = "pk"
                 if field.to_fields and field.to_fields[0] is not None:
-                    key = field.to_fields[0]
-                instance_kwargs = {key: default}
-                default = remote_field.model.objects.get(**instance_kwargs)
+                    default_key = field.to_fields[0]
+                default = remote_field.model.objects.get(**{default_key: default})
             return default
         elif field.name in self.attr_mapping:
             generator = self.attr_mapping[field.name]