Testing inference time, probably not a great idea...

model_server/extensions/ilastik/tests/test_ilastik.py

@@ -195,7 +195,7 @@ class TestIlastikPixelClassification(unittest.TestCase):
             channel=0,
         )
         self.assertTrue(result.success)
-        self.assertGreater(result.timer_results['inference'], 1.0)
+        self.assertGreater(result.timer_results['inference'], 0.1)
 
 class TestIlastikOverApi(TestServerBaseClass):
 

tests/test_api.py

@@ -15,7 +15,7 @@ class TestServerBaseClass(unittest.TestCase):
     def setUp(self) -> None:
         import uvicorn
         host = '127.0.0.1'
-        port = 5001 # TODO: defer to run_server script and conf file
+        port = 5001
 
         self.server_process = Process(
             target=uvicorn.run,