fixing raise-missing-from in remaining places

src/prefect/engine/serializers.py

@@ -88,10 +88,10 @@ def deserialize(self, value: bytes) -> Any:
             try:
                 # old versions of Core encoded pickles with base64
                 return cloudpickle.loads(base64.b64decode(value))
-            except Exception:
+            except Exception as e:
                 # if there's an error with the backwards-compatible step,
                 # reraise the original exception
-                raise exc
+                raise exc from e
 
 
 class JSONSerializer(Serializer):

src/prefect/tasks/airtable/__init__.py

@@ -3,7 +3,7 @@
 """
 try:
     from prefect.tasks.airtable.airtable import WriteAirtableRow, ReadAirtableRow
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.airtable` requires Prefect to be installed with the "airtable" extra.'
-    )
+    ) from err

src/prefect/tasks/aws/batch.py

@@ -90,8 +90,8 @@ def run(
                 jobDefinition=job_definition,
                 **batch_kwargs,
             )
-        except Exception:
-            raise FAIL(f"Failed to submit job '{job_name}' to AWS Batch.")
+        except Exception as e:
+            raise FAIL(f"Failed to submit job '{job_name}' to AWS Batch.") from e
 
         if not response.get("jobId"):
             raise FAIL(f"AWS Batch submit response contains no job ID: {response}")

src/prefect/tasks/aws/client_waiter.py

@@ -116,7 +116,9 @@ def run(
         try:
             waiter.wait(**waiter_kwargs)
         except WaiterError as e:
-            raise FAIL(f"AWS {client} waiter '{waiter_name}' failed with: {str(e)}")
+            raise FAIL(
+                f"AWS {client} waiter '{waiter_name}' failed with: {str(e)}"
+            ) from e
 
     @staticmethod
     def _load_prefect_waiter(
@@ -131,7 +133,7 @@ def _load_prefect_waiter(
                 waiter_model = WaiterModel(json.load(handle))
 
             return create_waiter_with_client(waiter_name, waiter_model, boto_client)
-        except Exception:
+        except Exception as err:
             raise ValueError(
                 f"Unable to load waiter '{waiter_name}' for AWS client '{client_str}'."
-            )
+            ) from err

src/prefect/tasks/azure/__init__.py

@@ -9,7 +9,7 @@
         CosmosDBReadItems,
         CosmosDBQueryItems,
     )
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.azure` requires Prefect to be installed with the "azure" extra.'
-    )
+    ) from err

src/prefect/tasks/azureml/__init__.py

@@ -16,7 +16,7 @@
         DatastoreUpload,
     )
 
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.azureml` requires Prefect to be installed with the "azure" extra.'
-    )
+    ) from err

src/prefect/tasks/dbt/__init__.py

@@ -4,5 +4,7 @@
 
 try:
     from prefect.tasks.dbt.dbt import DbtShellTask
-except ImportError:
-    raise ImportError("Using `prefect.tasks.dbt` requires dbt to be installed.")
+except ImportError as err:
+    raise ImportError(
+        "Using `prefect.tasks.dbt` requires dbt to be installed."
+    ) from err

src/prefect/tasks/dropbox/__init__.py

@@ -3,7 +3,7 @@
 """
 try:
     from prefect.tasks.dropbox.dropbox import DropboxDownload
-except ImportError:
+except ImportErro as err:
     raise ImportError(
         'Using `prefect.tasks.dropbox` requires Prefect to be installed with the "dropbox" extra.'
-    )
+    ) from err

src/prefect/tasks/gcp/bigquery.py

@@ -389,7 +389,7 @@ def run(
         except Exception as exception:
             for error in load_job.errors:
                 self.logger(error)
-            raise FAIL(exception)
+            raise FAIL(exception) from exception
         # remove unpickleable attributes
         load_job._client = None
         load_job._completion_lock = None

src/prefect/tasks/great_expectations/__init__.py

@@ -9,7 +9,7 @@
         RunGreatExpectationsValidation,
         RunGreatExpectationsCheckpoint,
     )
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.great_expectations` requires Prefect to be installed with the "ge" extra.'
-    )
+    ) from err

src/prefect/tasks/gsheets/__init__.py

@@ -6,7 +6,7 @@
         WriteGsheetRow,
         ReadGsheetRow,
     )
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.gsheets` requires Prefect to be installed with the "gsheets" extra.'
-    )
+    ) from err

src/prefect/tasks/kubernetes/__init__.py

@@ -41,7 +41,7 @@
         ReadNamespacedService,
         ReplaceNamespacedService,
     )
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.kubernetes` requires Prefect to be installed with the "kubernetes" extra.'
-    )
+    ) from err

src/prefect/tasks/postgres/__init__.py

@@ -9,7 +9,7 @@
         PostgresExecuteMany,
         PostgresFetch,
     )
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.postgres` requires Prefect to be installed with the "postgres" extra.'
-    )
+    ) from err

src/prefect/tasks/redis/__init__.py

@@ -5,7 +5,7 @@
 
 try:
     from prefect.tasks.redis.redis_tasks import RedisSet, RedisGet, RedisExecute
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.redis` requires Prefect to be installed with the "redis" extra.'
-    )
+    ) from err

src/prefect/tasks/rss/__init__.py

@@ -3,7 +3,7 @@
 """
 try:
     from prefect.tasks.rss.feed import ParseRSSFeed
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.rss` requires Prefect to be installed with the "rss" extra.'
-    )
+    ) from err

src/prefect/tasks/snowflake/__init__.py

@@ -5,7 +5,7 @@
 
 try:
     from prefect.tasks.snowflake.snowflake import SnowflakeQuery
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.snowflake` requires Prefect to be installed with the "snowflake" extra.'
-    )
+    ) from err

src/prefect/tasks/sql_server/__init__.py

@@ -9,7 +9,7 @@
         SqlServerExecuteMany,
         SqlServerFetch,
     )
-except ImportError:
+except ImportError as err:
     raise ImportError(
         'Using `prefect.tasks.sql_server` requires Prefect to be installed with the "sql_server" extra.'
-    )
+    ) from err

src/prefect/tasks/templates/jinja2.py

@@ -6,11 +6,11 @@
 
 try:
     from jinja2 import Template
-except ImportError:
+except ImportError as err:
     raise ImportError(
         "Using `prefect.tasks.templates.jinja2` requires Prefect to be installed "
         "with the 'templates' extra."
-    )
+    ) from err
 
 
 class JinjaTemplate(Task):