changed mock to unittest.mock

feathr-ai · Jun 27, 2022 · e20914b · e20914b
1 parent f39e07c
commit e20914b
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/feathr_project/docs/requirements.txt b/feathr_project/docs/requirements.txt
@@ -14,5 +14,4 @@ google>=3.0.0
 google-api-python-client>=2.41.0
 azure-keyvault-secrets
 confluent-kafka
-azure-core<=1.22.1
-mock
+azure-core<=1.22.1
diff --git a/feathr_project/setup.py b/feathr_project/setup.py
@@ -55,7 +55,6 @@
     ],
     tests_require=[
         'pytest',
-        'mock'
     ],
     entry_points={
         'console_scripts': ['feathr=feathrcli.cli:cli']

diff --git a/feathr_project/test/test_client_callback.py b/feathr_project/test/test_client_callback.py
@@ -1,6 +1,6 @@
 import os
 import asyncio
-import mock
+import unittest.mock as mock
 import time
 from subprocess import call
 from datetime import datetime, timedelta