diff --git a/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts b/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts index 5c490eb8eb6..66c6ebe504a 100644 --- a/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts +++ b/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts @@ -12,7 +12,7 @@ * Unit Tests for image resizing layer. */ -import {image, randomNormal, Rank, Tensor, tensor, zeros} from '@tensorflow/tfjs-core'; +import {image, Rank, Tensor, tensor, zeros} from '@tensorflow/tfjs-core'; // import {Shape} from '../../keras_format/common'; import {describeMathCPUAndGPU, expectTensorsClose} from '../../utils/test_utils'; @@ -86,7 +86,7 @@ describeMathCPUAndGPU('Resizing Layer', () => { // create a resizing layer with same shape as input const height = 64; const width = 32; - const numChannels = 3; + const numChannels = 1; const rangeArr = [...Array(height * width).keys()]; const inputArr = []; while(rangeArr.length) {