From 4eb9323f9c04e27225af50a92bd7a99cf54cc63f Mon Sep 17 00:00:00 2001
From: Oceania2018 <haiping008@gmail.com>
Date: Sat, 19 Dec 2020 10:01:27 -0600
Subject: [PATCH] fix unit test.

---
 src/TensorFlowNET.Core/Data/DatasetV2.cs | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/TensorFlowNET.Core/Data/DatasetV2.cs b/src/TensorFlowNET.Core/Data/DatasetV2.cs
index 763baa31..00b04edf 100644
--- a/src/TensorFlowNET.Core/Data/DatasetV2.cs
+++ b/src/TensorFlowNET.Core/Data/DatasetV2.cs
@@ -132,9 +132,7 @@ namespace Tensorflow
                     break;
                 }
 
-                yield return results.Length == 2
-                    ? (results[0], results[1])
-                    : (null, results[0]);
+                yield return (results[0], results.Length == 1 ? null : results[1]);
             }
         }