refactor

aniketmaurya · Aug 14, 2021 · 16322b6 · 16322b6
1 parent fedd539
commit 16322b6
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 14 deletions.
diff --git a/chitra/serve/base.py b/chitra/serve/base.py
@@ -65,7 +65,3 @@ def set_default_processor(self) -> DataProcessor:
                  {ModelServer.get_available_api_types()}"
             )
         return self.data_processor
-
-    @abc.abstractmethod
-    def run(self):
-        raise NotImplementedError
diff --git a/tests/serve/cloud/test_chalice_server.py b/tests/serve/cloud/test_chalice_server.py
@@ -14,7 +14,8 @@
 
 MODEL_PATH = "examples/assets/model.pth"
 if not os.path.exists(MODEL_PATH):
-    MODEL_PATH = "https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b0_ra-3dd342df.pth"
+    MODEL_PATH = "https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/\
+    efficientnet_b0_ra-3dd342df.pth"
 
 url = (
     "https://raw.githubusercontent.com/aniketmaurya/chitra/master/docs/assets/logo.png"

diff --git a/tests/serve/test_model_server.py b/tests/serve/test_model_server.py
@@ -1,7 +1,4 @@
-import pytest
-
 from chitra.serve import ModelServer
-from chitra.serve import constants as const
 
 
 def dummy_model(x: str):
@@ -11,9 +8,3 @@ def dummy_model(x: str):
 def test_get_available_api_types():
     model_server = ModelServer.get_available_api_types()
     assert isinstance(model_server, list)
-
-
-def test_model_server_run():
-    model_server = ModelServer(api_type=const.TXT_CLF, model=dummy_model)
-    with pytest.raises(NotImplementedError):
-        model_server.run()