From cef741527d7a48e45ed065f92b0dfc9fb4119a72 Mon Sep 17 00:00:00 2001 From: Christoph Buelter Date: Wed, 9 Dec 2020 15:49:31 +0100 Subject: [PATCH] Shorten kwargs setup --- model_bakery/baker.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/model_bakery/baker.py b/model_bakery/baker.py index fa340e8f..37a349f1 100644 --- 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/ 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]