huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 04:10:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix a weird automatic RustRover change

Browse Source
This commit is contained in:
b1rtek
2024-06-04 21:13:30 +02:00
parent a5b81e2c02
commit c441716bd2
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
candle-onnx/tests/ops.rs
View File

@ -2711,7 +2711,7 @@ fn test_ceil() -> Result<()> {
// "ArgMin" // "ArgMin"
#[test] #[test]
fn test_argmin() -> std::result::Result<()> { fn test_argmin() -> Result<()> {
// tests from https://github.com/onnx/onnx/blob/main/docs/Operators.md#examples-7 // tests from https://github.com/onnx/onnx/blob/main/docs/Operators.md#examples-7
// default_axes_keepdims // default_axes_keepdims
test( test(
@ -2790,7 +2790,7 @@ fn test_argmin() -> std::result::Result<()> {
None, None,
&[[2i64], [1i64], [3i64], [1i64]], &[[2i64], [1i64], [3i64], [1i64]],
)?; )?;
fn test(data: impl NdArray, axis: Option<i64>, keepdims: Option<i64>, select_last_index: Option<i64>, expected: impl NdArray) -> std::result::Result<()> { fn test(data: impl NdArray, axis: Option<i64>, keepdims: Option<i64>, select_last_index: Option<i64>, expected: impl NdArray) -> Result<()> {
let att_axis = AttributeProto { let att_axis = AttributeProto {
name: "axis".to_string(), name: "axis".to_string(),
ref_attr_name: "axis".to_string(), ref_attr_name: "axis".to_string(),
@ -2907,7 +2907,7 @@ fn test_argmin() -> std::result::Result<()> {
// "ArgMax" // "ArgMax"
#[test] #[test]
fn test_argmax() -> std::result::Result<()> { fn test_argmax() -> Result<()> {
// tests from https://github.com/onnx/onnx/blob/main/docs/Operators.md#examples-6 // tests from https://github.com/onnx/onnx/blob/main/docs/Operators.md#examples-6
// default_axes_keepdims // default_axes_keepdims
test( test(
@ -2986,7 +2986,7 @@ fn test_argmax() -> std::result::Result<()> {
None, None,
&[[0i64], [2i64], [0i64], [1i64]], &[[0i64], [2i64], [0i64], [1i64]],
)?; )?;
fn test(data: impl NdArray, axis: Option<i64>, keepdims: Option<i64>, select_last_index: Option<i64>, expected: impl NdArray) -> std::result::Result<()> { fn test(data: impl NdArray, axis: Option<i64>, keepdims: Option<i64>, select_last_index: Option<i64>, expected: impl NdArray) -> Result<()> {
let att_axis = AttributeProto { let att_axis = AttributeProto {
name: "axis".to_string(), name: "axis".to_string(),
ref_attr_name: "axis".to_string(), ref_attr_name: "axis".to_string(),