Hacktoberfest 2026: los issues que los mantenedores marcaron para octubre, abiertos y aptos para principiantes. Explorar issues de Hacktoberfest

Incorrect shape assertion in test_decode_prefixes_device_handling

Abierto
#23 0 comentarios 0 reacciones 0 asignados Ver en GitHub

@samuelstevens ya está trabajando en esto.

Desde el 14/10/2025.

  • #24 de @samuelstevens — abierto

Evaluación

Dificultad
1/5
Tiempo estimado
Menos de una hora
Aptitud para principiantes
52/100
Tipo de issue
Error
Claridad
Bien especificado
Estado de actividad
Estancado
Stack tecnológico
python

Línea de trabajo

Abre test/test_nn_objectives.py e inspecciona test_decode_prefixes_device_handling alrededor de la línea 491, incluida la forma esperada de decode. Ejecuta el test en un sistema con CUDA y actualiza la assertion para que coincida con el orden documentado (batch, n_prefixes, d_model).

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

Bug Description

The test test_decode_prefixes_device_handling in tests/test_nn_objectives.py has an incorrect shape assertion on line 491.

Current Code

f_x = torch.randn(2, 32).cuda()
prefixes = torch.tensor([8, 16, 32])
x_hats = sae.decode(f_x, prefixes=prefixes)
assert x_hats.shape == (3, 2, 16)  # BUG: Wrong shape!

Expected Behavior

The decode method returns tensors with shape (batch, n_prefixes, d_model). Given:

  • batch_size = 2
  • n_prefixes = 3
  • d_model = 16

The correct shape should be (2, 3, 16) not (3, 2, 16).

Impact

This test is currently skipped on systems without CUDA, so the bug hasn't been caught by CI. However, on CUDA-enabled systems, this test would fail.

Fix

Change line 491 from:

assert x_hats.shape == (3, 2, 16)

to:

assert x_hats.shape == (2, 3, 16)
Lenguaje dominante
Python
Estrellas
67
Forks
12
Métricas de merge de PR
Sin PR fusionados en 30 d

Preparar el entorno

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de Imageomics/saev

Todos los issues de Imageomics/saev

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.