Refactor test_bigbench_reasoning to improve input structure and clarity

Neuro-Flex · Dec 25, 2024 · 5575ade · 5575ade
1 parent 6ae6871
commit 5575ade
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/tests/benchmarks/__pycache__/test_bigbench_reasoning.cpython-310-pytest-8.3.4.pyc b/tests/benchmarks/__pycache__/test_bigbench_reasoning.cpython-310-pytest-8.3.4.pyc
diff --git a/tests/benchmarks/test_bigbench_reasoning.py b/tests/benchmarks/test_bigbench_reasoning.py
@@ -123,12 +123,11 @@ def test_meta_learning(self, device, consciousness_model):
                             'textual': input_embedding,
                             'visual': torch.zeros(1, 64, 512, device=device),  # Added dummy 'visual' input with matching sequence length
                             'state': state
-                        consciousness_threshold=0.1 + i*0.3  # Increase threshold with complexity
-                    )
-                        consciousness_threshold=0.1 + i*0.3  # Increase threshold with complexity
+                        },
+                        consciousness_threshold=0.1 + i*0.3,  # Increase threshold with complexity
                     )
                     phi_values.append(metrics['phi'])
-                    last_state = output.squeeze(0)
+                    last_state = output
 
             # Test adaptation capability with more lenient thresholds
             phi_tensor = torch.cat(phi_values)