diff --git a/src/dsdk/persistor.py b/src/dsdk/persistor.py index a892d27..f4f385e 100644 --- a/src/dsdk/persistor.py +++ b/src/dsdk/persistor.py @@ -171,10 +171,8 @@ def dry_run( parameters: Dict[str, Any], exceptions: Tuple = (), ): - """Execute sql found in asse with dry_run parameter set to 1.""" + """Execute sql found in asset with dry_run.""" logger.info(self.ON) - parameters = parameters.copy() - parameters["dry_run"] = 1 errors = [] for path, query in self.sql(): logger.info(self.DRY_RUN, path) @@ -192,12 +190,12 @@ def dry_run_query( parameters, exceptions: Tuple = (), ) -> Optional[Exception]: - """Dry run query.""" + """Dry run query with dry_run parameter set to 1.""" with self.rollback() as cur: rendered = self.render_without_keys( cur, query, - parameters, + {**parameters, "dry_run": 1}, ) with NamedTemporaryFile("w", delete=False, suffix=".sql") as fout: fout.write(rendered)