Obey YAML file in reload pk check

coursera · Sep 8, 2015 · a76e2e3 · a76e2e3
1 parent eff48b9
commit a76e2e3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dataduct/steps/reload.py b/dataduct/steps/reload.py
@@ -13,8 +13,10 @@ def __init__(self, **kwargs):
         Args:
             **kwargs(optional): Keyword arguments directly passed to base class
         """
+
         # Do not enforce pkey for reload so we can surface data quality issues.
-        kwargs['enforce_primary_key'] = False
+        if 'enforce_primary_key' not in kwargs:
+            kwargs['enforce_primary_key'] = False
         super(ReloadStep, self).__init__(**kwargs)
 
     @classmethod