From 71e874e1769d2ee69f69aa6f4749129933144052 Mon Sep 17 00:00:00 2001
From: shiyu22 <shiyu.chen@zilliz.com>
Date: Wed, 15 Dec 2021 21:03:45 +0800
Subject: [PATCH] Update

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
---
 __init__.py         | 12 ++++++------
 pytorch/__init__.py |  9 +++++++++
 2 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/__init__.py b/__init__.py
index efca5a9..fffb2ea 100644
--- a/__init__.py
+++ b/__init__.py
@@ -12,10 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
+# import os
 
-# For requirements.
-try:
-    import efficientnet_pytorch
-except ModuleNotFoundError:
-    os.system('pip install efficientnet_pytorch')
+# # For requirements.
+# try:
+#     import efficientnet_pytorch
+# except ModuleNotFoundError:
+#     os.system('pip install efficientnet_pytorch')
diff --git a/pytorch/__init__.py b/pytorch/__init__.py
index 37f5bd7..ed43fad 100644
--- a/pytorch/__init__.py
+++ b/pytorch/__init__.py
@@ -11,3 +11,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+
+import os
+
+# For requirements.
+try:
+    import efficientnet_pytorch
+except ModuleNotFoundError:
+    os.system('pip install efficientnet_pytorch')
\ No newline at end of file