Fix assert in TextIOIT

apache · Oct 9, 2024 · d51f36c · d51f36c
1 parent 42cad40
commit d51f36c
Showing 1 changed file with 1 addition and 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
@@ -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 {