Update tfjs-layers/src/layers/preprocessing/preprocessing_utils.ts

Co-authored-by: Matthew Soulanille <matthew@soulanille.net>
CodeSmithDSMLProjects · Oct 4, 2022 · 7222f30 · 7222f30
1 parent 7460615
commit 7222f30
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/tfjs-layers/src/layers/preprocessing/preprocessing_utils.ts b/tfjs-layers/src/layers/preprocessing/preprocessing_utils.ts
@@ -52,15 +52,7 @@ export function encodeCategoricalInputs(inputs: Tensor|Tensor[],
 
   const binaryOutput = ['multiHot', 'oneHot'].includes(outputMode);
 
-  let denseBincountInput: Tensor1D | Tensor2D;
-
-  if(input.rank === 1) {
-    denseBincountInput = input as Tensor1D;
-  }
-
-  if(input.rank === 2) {
-    denseBincountInput = input as Tensor2D;
-  }
+  const denseBincountInput = input as Tensor1D | Tensor2D;
 
   let binCounts: Tensor1D | Tensor2D;