nit

Signed-off-by: Yee Hing Tong <[email protected]>
flyteorg · Aug 30, 2021 · c478308 · c478308
1 parent b371d0f
commit c478308
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/flytekit/unit/remote/test_remote.py b/tests/flytekit/unit/remote/test_remote.py
@@ -1,6 +1,8 @@
 import random
 
 import pytest
+from mock import MagicMock, patch
+
 from flytekit.models import common as common_models
 from flytekit.models.admin.workflow import Workflow
 from flytekit.models.core.identifier import (
@@ -17,7 +19,6 @@
 from flytekit.models.types import LiteralType, SimpleType
 from flytekit.remote import FlyteWorkflow
 from flytekit.remote.remote import FlyteRemote
-from mock import MagicMock, patch
 
 CLIENT_METHODS = {
     ResourceType.WORKFLOW: "list_workflows_paginated",