From 7e3bcb3dd62198be25ff8115b7707893365c97cd Mon Sep 17 00:00:00 2001 From: Jake LoRocco Date: Tue, 26 Aug 2025 09:05:46 -0400 Subject: [PATCH 1/2] skip rich doc test that takes too much memory --- test/stdlib_basics/test_richdocument.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/stdlib_basics/test_richdocument.py b/test/stdlib_basics/test_richdocument.py index 8809b1a0..6046ea96 100644 --- a/test/stdlib_basics/test_richdocument.py +++ b/test/stdlib_basics/test_richdocument.py @@ -93,6 +93,7 @@ def test_empty_table(): assert table is None, "table should be empty when supplied string is empty" +@pytest.mark.skip # Test requires too much memory for smaller machines. def test_richdocument_generation(rd: RichDocument): m = mellea.start_session(backend_name="hf") response = m.chat(rd.to_markdown()[:500] + "\nSummarize the provided document.") From 56b6a229b468138851fbd5e05a883e27ecf576e6 Mon Sep 17 00:00:00 2001 From: Jake LoRocco Date: Tue, 26 Aug 2025 09:07:02 -0400 Subject: [PATCH 2/2] remove unused session functions --- mellea/stdlib/session.py | 25 ------------------------- 1 file changed, 25 deletions(-) diff --git a/mellea/stdlib/session.py b/mellea/stdlib/session.py index 0d2d6f97..0250b768 100644 --- a/mellea/stdlib/session.py +++ b/mellea/stdlib/session.py @@ -414,31 +414,6 @@ def validate( return rvs - def req(self, *args, **kwargs): - """Shorthand for Requirement.__init__(...).""" - return req(*args, **kwargs) - - def check(self, *args, **kwargs): - """Shorthand for Requirement.__init__(..., check_only=True).""" - return check(*args, **kwargs) - - def load_default_aloras(self): - """Loads the default Aloras for this model, if they exist and if the backend supports.""" - from mellea.backends.huggingface import LocalHFBackend - - if self.backend.model_id == IBM_GRANITE_3_2_8B and isinstance( - self.backend, LocalHFBackend - ): - from mellea.backends.aloras.huggingface.granite_aloras import ( - add_granite_aloras, - ) - - add_granite_aloras(self.backend) - return - self._session_logger.warning( - "This model/backend combination does not support any aloras." - ) - def genslot( self, gen_slot: Component,