Please, linter gods, pleaaaase stop hating me

BrainJS · Jun 17, 2024 · 83574f6 · 83574f6
1 parent a21c387
commit 83574f6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/neural-network.ts b/src/neural-network.ts
@@ -323,7 +323,12 @@ export class NeuralNetwork<
   constructor(
     options: Partial<INeuralNetworkOptions & INeuralNetworkTrainOptions> = {}
   ) {
-    this.options = { ...this.options, ...options };
+    this.options.binaryThresh = options.binaryThresh ?? 0.5;
+    this.options.hiddenLayers = options.hiddenLayers ?? [];
+    this.options.inputSize = options.inputSize ?? 1;
+    this.options.loss = options.loss ?? loss;
+    this.options.outputSize = options.outputSize ?? options.inputSize ?? 1;
+    this.options.ramSize = options.ramSize ?? 1;
     this.updateTrainingOptions(options);
 
     const { inputSize, hiddenLayers, outputSize } = this.options;

diff --git a/src/utilities/to-svg.ts b/src/utilities/to-svg.ts
@@ -467,7 +467,9 @@ export function toSVG<
   // Get network size array for NeuralNetwork or NeuralNetworkGPU
   let sizes: number[] = [];
   if (net instanceof NeuralNetwork || net instanceof NeuralNetworkGPU) {
-    sizes = getNeuralNetworkSizes(net);
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    // @ts-expect-error
+    sizes = getNeuralNetworkSizes((net as unknown) as NeuralNetwork);
   }
   // get network size for Recurrent
   else if (net instanceof Recurrent) {