def METHOD_NAME():
    @dace.program
    def test_argmin_result(A: dace.float64[10, 5, 3]):
        return np.argmin(A, axis=1, result_type=dace.int64)
    res = test_argmin_result(np.random.rand(10, 5, 3))
    assert res.dtype == np.int64
    @dace.program
    def test_argmin_result(A: dace.float64[10, 5, 3]):
        return np.argmin(A, axis=1)
    res = test_argmin_result(np.random.rand(10, 5, 3))
    assert res.dtype == np.int32