From d51f36cbfd67d03dd65292e314d4a07a900fcb59 Mon Sep 17 00:00:00 2001
From: Yi Hu <yathu@google.com>
Date: Wed, 9 Oct 2024 10:07:55 -0400
Subject: [PATCH] Fix assert in TextIOIT

---
 .../src/test/java/org/apache/beam/sdk/io/text/TextIOIT.java     | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdks/java/io/file-based-io-tests/src/test/java/org/apache/beam/sdk/io/text/TextIOIT.java b/sdks/java/io/file-based-io-tests/src/test/java/org/apache/beam/sdk/io/text/TextIOIT.java
index ecdde5cbc8fe7..e50a8aba41624 100644
--- a/sdks/java/io/file-based-io-tests/src/test/java/org/apache/beam/sdk/io/text/TextIOIT.java
+++ b/sdks/java/io/file-based-io-tests/src/test/java/org/apache/beam/sdk/io/text/TextIOIT.java
@@ -157,7 +157,7 @@ public void writeThenReadAll() {
 
     Set<String> sources = Lineage.query(result.metrics(), Lineage.Type.SOURCE);
     Set<String> sinks = Lineage.query(result.metrics(), Lineage.Type.SINK);
-    if (numShards <= 100) {
+    if (numShards != null && numShards <= 100) {
       // both should be the full files, if supported by the runner
       assertEquals(sources, sinks);
     } else {