Skip to content
Snippets Groups Projects
Commit 356069a6 authored by Christopher Randolph Rhodes's avatar Christopher Randolph Rhodes
Browse files

Merged in usage of session module global object instead of singleton class;...

Merged in usage of session module global object instead of singleton class; still working through tests
parent bcc66f86
No related branches found
No related tags found
No related merge requests found
from fastapi import APIRouter, HTTPException
from model_server.base.session import session
from model_server.base.validators import validate_workflow_inputs
from base.session import session
from base.validators import validate_workflow_inputs
from model_server.extensions.ilastik import models as ilm
from model_server.base.models import ParameterExpectedError
from model_server.extensions.ilastik.workflows import infer_px_then_ob_model
from extensions.ilastik import models as ilm
from extensions.ilastik.workflows import infer_px_then_ob_model
router = APIRouter(
prefix='/ilastik',
......
......@@ -3,7 +3,6 @@ import pathlib
from pydantic import BaseModel
import unittest
import base.session
from base.models import DummySemanticSegmentationModel
from base.session import session
......@@ -50,12 +49,6 @@ class TestGetSessionObject(unittest.TestCase):
self.assertTrue(logfile2.exists())
self.assertEqual(logfile1, logfile2, 'Reimporting session incorrectly creates new logfile')
from model_server.base.session import session as session3
self.assertEqual(session1, session3)
logfile3 = session3.logfile
self.assertTrue(logfile3.exists())
self.assertEqual(logfile1, logfile3, 'Reimporting session incorrectly creates new logfile')
def test_log_warning(self):
msg = 'A test warning'
self.sesh.log_info(msg)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment