@@ -1356,7 +1356,6 @@ abstract class TransformWithStateTest extends StateStoreMetricsTest
1356
1356
TransformWithStateSuiteUtils .NUM_SHUFFLE_PARTITIONS .toString) {
1357
1357
withTempDir { checkpointDir =>
1358
1358
// When Avro is used, we want to set the StructFields to nullable
1359
- val shouldBeNullable = usingAvroEncoding()
1360
1359
val metadataPathPostfix = " state/0/_stateSchema/default"
1361
1360
val stateSchemaPath = new Path (checkpointDir.toString,
1362
1361
s " $metadataPathPostfix" )
@@ -1367,15 +1366,15 @@ abstract class TransformWithStateTest extends StateStoreMetricsTest
1367
1366
val schema0 = StateStoreColFamilySchema (
1368
1367
" countState" , 0 ,
1369
1368
keySchema, 0 ,
1370
- new StructType ().add(" value" , LongType , nullable = shouldBeNullable ),
1369
+ new StructType ().add(" value" , LongType , nullable = true ),
1371
1370
Some (NoPrefixKeyStateEncoderSpec (keySchema)),
1372
1371
None
1373
1372
)
1374
1373
val schema1 = StateStoreColFamilySchema (
1375
1374
" listState" , 0 ,
1376
1375
keySchema, 0 ,
1377
1376
new StructType ()
1378
- .add(" id" , LongType , nullable = shouldBeNullable )
1377
+ .add(" id" , LongType , nullable = true )
1379
1378
.add(" name" , StringType ),
1380
1379
Some (NoPrefixKeyStateEncoderSpec (keySchema)),
1381
1380
None
@@ -1398,7 +1397,7 @@ abstract class TransformWithStateTest extends StateStoreMetricsTest
1398
1397
val schema3 = StateStoreColFamilySchema (
1399
1398
" $rowCounter_listState" , 0 ,
1400
1399
keySchema, 0 ,
1401
- new StructType ().add(" count" , LongType , nullable = shouldBeNullable ),
1400
+ new StructType ().add(" count" , LongType , nullable = true ),
1402
1401
Some (NoPrefixKeyStateEncoderSpec (keySchema)),
1403
1402
None
1404
1403
)
0 commit comments