* Re-enable skipped test and fix the hidden state shape issue * Actually fix the bug instead of just doing something wrong
is_pipeline_test_to_skip