From 38c11c24ad237bc1b07986ac2474b7a5c0de4b8f Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Wed, 3 Jul 2024 12:59:51 +0200
Subject: [PATCH] Still trying to quiet ilastik and its dependencies on main
log
---
model_server/extensions/ilastik/models.py | 11 +++++++++--
tests/test_ilastik.py | 1 +
2 files changed, 10 insertions(+), 2 deletions(-)
diff --git a/model_server/extensions/ilastik/models.py b/model_server/extensions/ilastik/models.py
index af7844f3..5e7a0a75 100644
--- a/model_server/extensions/ilastik/models.py
+++ b/model_server/extensions/ilastik/models.py
@@ -1,6 +1,8 @@
import json
+from logging import getLogger
import os
from pathlib import Path
+import warnings
import numpy as np
from pydantic import BaseModel
@@ -43,8 +45,13 @@ class IlastikModel(Model):
super().__init__(autoload, params)
def load(self):
- from ilastik import app
- from ilastik.applets.dataSelection.opDataSelection import PreloadedArrayDatasetInfo
+ # suppress warnings when loading ilastik app
+ getLogger('ilastik.app').setLevel('ERROR')
+
+ with warnings.catch_warnings():
+ warnings.filterwarnings('ignore', category=DeprecationWarning)
+ from ilastik import app
+ from ilastik.applets.dataSelection.opDataSelection import PreloadedArrayDatasetInfo
self.PreloadedArrayDatasetInfo = PreloadedArrayDatasetInfo
diff --git a/tests/test_ilastik.py b/tests/test_ilastik.py
index 66a3c0c2..a31dbf88 100644
--- a/tests/test_ilastik.py
+++ b/tests/test_ilastik.py
@@ -11,6 +11,7 @@ from model_server.base.roiset import _get_label_ids, RoiSet, RoiSetMetaParams
from model_server.base.workflows import classify_pixels
from tests.test_api import TestServerBaseClass
+
def _random_int(*args):
return np.random.randint(0, 2 ** 8, size=args, dtype='uint8')
--
GitLab