fixup! Add dynamic task mapping into TaskSDK runtime

Author: ashb

tests_common/pytest_plugin.py

@@ -544,8 +544,7 @@ def skip_db_test(item):
         # also automatically skip tests marked with `backend` marker as they are implicitly
         # db tests
         pytest.skip(
-            f"The test is skipped as it is DB test "
-            f"and --skip-db-tests is flag is passed to pytest. {item}"
+            f"The test is skipped as it is DB test and --skip-db-tests is flag is passed to pytest. {item}"
         )
 
 
@@ -876,6 +875,7 @@ def __exit__(self, type, value, traceback):
                     sdm._SerializedDagModel__data_cache = (
                         self.serialized_model._SerializedDagModel__data_cache
                     )
+                    sdm._data = self.serialized_model._data
                     self.serialized_model = sdm
                 else:
                     self.session.merge(self.serialized_model)