issue 676

Signed-off-by: Xavier Dupre <xadupre@microsoft.com>
onnx · Jan 23, 2024 · 6948f49 · 6948f49
1 parent cb84112
commit 6948f49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/xgboost/test_xgboost_issues.py b/tests/xgboost/test_xgboost_issues.py
@@ -42,7 +42,7 @@ def test_issue_676(self):
             onnx_model.SerializeToString(), providers=["CPUExecutionProvider"]
         )
         got = sess.run(None, {"float_input": X.astype(np.float32)})
-        self.assertEqual(got[0].shape, (100, 1))
+        self.assertEqual(got[0].shape, (100, 210))
 
 
 if __name__ == "__main__":