From 16927ba97a8ed54e66b012f020ebe9e6a73bb8f6 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Thu, 6 Jun 2024 11:29:31 +0200
Subject: [PATCH] Testing inference time, probably not a great idea...
---
model_server/extensions/ilastik/tests/test_ilastik.py | 2 +-
tests/test_api.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/model_server/extensions/ilastik/tests/test_ilastik.py b/model_server/extensions/ilastik/tests/test_ilastik.py
index 1907459d..420cd0a0 100644
--- a/model_server/extensions/ilastik/tests/test_ilastik.py
+++ b/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):
diff --git a/tests/test_api.py b/tests/test_api.py
index 9cb56c68..6e336a9c 100644
--- a/tests/test_api.py
+++ b/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,
--
GitLab