Fix test (#2806)

### Changes

Revert some changes of the  `tests/shared/helpers.py` file

### Reason for changes
#2218


![image](https://github.com/user-attachments/assets/c2140e7a-8216-4d3f-a3e2-80ea66270e27)


### Related tickets

N/A

### Tests

N/A

tests/shared/helpers.py

@@ -201,8 +201,8 @@ def compare_stats(expected: Dict[str, np.ndarray], actual: Dict[str, np.ndarray]
         actual_stats = actual[ref_node_name]
         for param_name, ref_param in ref_stats.items():
             actual_param = actual_stats.get(param_name)
-        assert np.array(ref_param).shape == np.array(actual_param).shape
-        assert np.allclose(ref_param, actual_param, atol=1e-5)
+            assert np.array(ref_param).shape == np.array(actual_param).shape
+            assert np.allclose(ref_param, actual_param, atol=1e-5)
 
 
 def get_python_executable_with_venv(venv_path: Path) -> str: