Fix import

PrefectHQ · Oct 10, 2022 · cd5ef15 · cd5ef15
1 parent c1de8bf
commit cd5ef15
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_tasks.py b/tests/test_tasks.py
@@ -9,6 +9,7 @@
 import pytest
 
 from prefect import flow, get_run_logger, tags
+from prefect.blocks.core import Block
 from prefect.context import PrefectObjectRegistry, TaskRunContext, get_run_context
 from prefect.deprecated.data_documents import DataDocument
 from prefect.engine import get_state_for_result