Skip to content

Commit 8c69bc1

Browse files
jeremymanningclaude
andcommitted
Fix test parameters for ParticleFilter.generate()
- Fixed incorrect parameter names in tests - Changed n_steps to max_new_tokens to match API - All file operation tests passing (4/4) Part of issue #4 test suite fixes. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent 2e8639f commit 8c69bc1

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

code/tests/test_cleanup_real.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -144,8 +144,7 @@ def test_cleanup_preserves_model_functionality(self):
144144
)
145145

146146
# Real generation
147-
pf.initialize("Test prompt")
148-
particles = pf.generate(n_steps=10) # Use n_steps instead of max_length
147+
particles = pf.generate("Test prompt", max_new_tokens=10)
149148

150149
# Verify real output
151150
assert len(particles) == 2
@@ -159,8 +158,7 @@ def test_visualizer_after_cleanup(self):
159158
n_particles=3,
160159
device="cpu"
161160
)
162-
pf.initialize("Hello")
163-
particles = pf.generate(n_steps=5) # Use n_steps
161+
particles = pf.generate("Hello", max_new_tokens=5)
164162

165163
# Initialize visualizer with tokenizer
166164
visualizer = TokenSequenceVisualizer(pf.tokenizer)

0 commit comments

Comments
 (0)