diff --git a/tfjs-layers/src/layers/preprocessing/image_resizing.ts b/tfjs-layers/src/layers/preprocessing/image_resizing.ts index ecd01f809e7..9ebb808d7ca 100644 --- a/tfjs-layers/src/layers/preprocessing/image_resizing.ts +++ b/tfjs-layers/src/layers/preprocessing/image_resizing.ts @@ -26,7 +26,7 @@ export declare interface ResizingArgs extends LayerArgs { height: number; width: number; interpolation?: InterpolationType; // default = 'bilinear'; - cropToAspectRatio?: boolean; // default = false; + cropToAspectRatio?: boolean; // default = false; } /** diff --git a/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts b/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts index eab494fd88a..51ae2916683 100644 --- a/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts +++ b/tfjs-layers/src/layers/preprocessing/image_resizing_test.ts @@ -17,7 +17,7 @@ import {image, Rank, Tensor, tensor, zeros, range, reshape} from '@tensorflow/tf // import {Shape} from '../../keras_format/common'; import {describeMathCPUAndGPU, expectTensorsClose} from '../../utils/test_utils'; -import {Resizing} from './image_resizing'; +import {Resizing, ResizingArgs} from './image_resizing'; describeMathCPUAndGPU('Resizing Layer', () => { it('Check if output shape matches specifications', () => { @@ -110,7 +110,7 @@ describeMathCPUAndGPU('Resizing Layer', () => { const incorrectArgs = {height, width, interpolation}; const expectedError = `Invalid interpolation parameter: ${interpolation} is not implemented`; - expect(() => new Resizing(incorrectArgs)).toThrowError(expectedError); + expect(() => new Resizing(incorrectArgs as ResizingArgs)).toThrowError(expectedError); }); it('Config holds correct name', () => {