Update

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>

__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')

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')