From 3791cd65857c8ec899203c0dfabd0f3eea5ef342 Mon Sep 17 00:00:00 2001
From: kshtiijrajsharma <skshitizraj@gmail.com>
Date: Mon, 25 Mar 2024 09:00:19 +0545
Subject: [PATCH] (ci) unit-test : remove efficientnet

---
 .github/workflows/unit-test.yml | 2 +-
 tests/test_predict.py           | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/unit-test.yml b/.github/workflows/unit-test.yml
index b516b4e..db41203 100644
--- a/.github/workflows/unit-test.yml
+++ b/.github/workflows/unit-test.yml
@@ -20,7 +20,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install tensorflow==2.11.0 efficientnet==1.0.8
+          pip install tensorflow==2.11.0
           pip install -e .
 
       - name: Run tests
diff --git a/tests/test_predict.py b/tests/test_predict.py
index f244b03..ef8cb72 100644
--- a/tests/test_predict.py
+++ b/tests/test_predict.py
@@ -3,7 +3,7 @@
 import tempfile
 import unittest
 
-import efficientnet.keras as efn
+# import efficientnet.keras as efn
 import requests
 
 from predictor import predict