Replace imports

Author: tconley1428

tests/worker/test_workflow.py

@@ -15,13 +15,15 @@
 import typing
 import uuid
 from abc import ABC, abstractmethod
+from contextlib import contextmanager
 from dataclasses import dataclass
 from datetime import datetime, timedelta, timezone
 from enum import IntEnum
 from functools import partial
 from typing import (
     Any,
     Awaitable,
+    Callable,
     Dict,
     List,
     Mapping,