feat: update lumina system prompt handling

2026-04-06 13:47:06 +00:00 · 2025-06-29 21:33:09 +09:00
parent 52d13373c0
commit 935e0037dc
6 changed files with 11 additions and 17 deletions
--- a/tests/library/test_strategy_lumina.py
+++ b/tests/library/test_strategy_lumina.py
@@ -126,13 +126,12 @@ def test_lumina_text_encoder_outputs_caching_strategy():

        # Create a mock class for ImageInfo
        class MockImageInfo:
-            def __init__(self, caption, system_prompt, cache_path):
+            def __init__(self, caption, cache_path):
                self.caption = caption
-                self.system_prompt = system_prompt
                self.text_encoder_outputs_npz = cache_path

        # Create a sample input info
-        image_info = MockImageInfo("Test caption", "", cache_file)
+        image_info = MockImageInfo("Test caption", cache_file)

        # Simulate a batch
        batch = [image_info]