diff --git a/eng/Versions.props b/eng/Versions.props index 6bbd721e2d..6d73dd895d 100644 --- a/eng/Versions.props +++ b/eng/Versions.props @@ -23,7 +23,7 @@ 3.10.1 2.2.3 2.1.0 - 1.5.2 + 1.6.0 0.0.0.9 2.1.3 4.5.0 diff --git a/src/Microsoft.ML.OnnxTransformer/OnnxTransform.cs b/src/Microsoft.ML.OnnxTransformer/OnnxTransform.cs index c0900b5ca6..df94857931 100644 --- a/src/Microsoft.ML.OnnxTransformer/OnnxTransform.cs +++ b/src/Microsoft.ML.OnnxTransformer/OnnxTransform.cs @@ -768,7 +768,7 @@ public NamedOnnxValue GetNamedOnnxValue() /// | Does this estimator need to look at the data to train its parameters? | No | /// | Input column data type | Known-sized vector of or types | /// | Output column data type | As specified by the ONNX model | - /// | Required NuGet in addition to Microsoft.ML | Microsoft.ML.OnnxTransformer (always), either Microsoft.ML.OnnxRuntime 1.5.2 (for CPU processing) or Microsoft.ML.OnnxRuntime.Gpu 1.5.2 (for GPU processing if GPU is available) | + /// | Required NuGet in addition to Microsoft.ML | Microsoft.ML.OnnxTransformer (always), either Microsoft.ML.OnnxRuntime 1.6.0 (for CPU processing) or Microsoft.ML.OnnxRuntime.Gpu 1.6.0 (for GPU processing if GPU is available) | /// | Exportable to ONNX | No | /// /// To create this estimator use the following APIs: diff --git a/src/Microsoft.ML.OnnxTransformer/OnnxTypeParser.cs b/src/Microsoft.ML.OnnxTransformer/OnnxTypeParser.cs index f2b4a943d1..1a2b1420e4 100644 --- a/src/Microsoft.ML.OnnxTransformer/OnnxTypeParser.cs +++ b/src/Microsoft.ML.OnnxTransformer/OnnxTypeParser.cs @@ -267,7 +267,13 @@ private class CastHelper public static IEnumerable CastOnnxSequenceToIEnumerable(IEnumerable o, Func caster) { - return o.Select(v => (TDst)caster(v)); + // Since now we're disposing the NamedOnnxValue objects + // after running inference on each output, we need + // to copy (enumerate) the output through ".ToList()" + // else, if our users try the keep the past sequence + // outputs of their OnnxTransformer, they would + // end up with empty sequences. + return o.Select(v => (TDst)caster(v)).ToList(); } }