Set pathOrIOHandler to IOHandler

Morgan Laco · Morgan Laco · commit 6841bb7d3c61 · 2019-08-01T10:26:45.000-04:00
diff --git a/src/index.ts b/src/index.ts
@@ -43,7 +43,7 @@ export class NSFWJS {
   public endpoints: string[]
 
   private options: nsfwjsOptions
-  private pathOrIOHandler: string
+  private pathOrIOHandler: string | IOHandler
   private model: tf.LayersModel
   private intermediateModels: { [layerName: string]: tf.LayersModel } = {}
 
@@ -59,6 +59,7 @@ export class NSFWJS {
     if (typeof modelPathBaseOrIOHandler === 'string') {
       this.pathOrIOHandler = `${modelPathBaseOrIOHandler}model.json`
     } else {
+      this.pathOrIOHandler = modelPathBaseOrIOHandler
     }
   }
 

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@ export class NSFWJS {`
`43`	`43`	`public endpoints: string[]`
`44`	`44`
`45`	`45`	`private options: nsfwjsOptions`
`46`		`- private pathOrIOHandler: string`
	`46`	`+ private pathOrIOHandler: string \| IOHandler`
`47`	`47`	`private model: tf.LayersModel`
`48`	`48`	`private intermediateModels: { [layerName: string]: tf.LayersModel } = {}`
`49`	`49`
`@@ -59,6 +59,7 @@ export class NSFWJS {`
`59`	`59`	`if (typeof modelPathBaseOrIOHandler === 'string') {`
`60`	`60`	this.pathOrIOHandler = `${modelPathBaseOrIOHandler}model.json`
`61`	`61`	`} else {`
	`62`	`+ this.pathOrIOHandler = modelPathBaseOrIOHandler`
`62`	`63`	`}`
`63`	`64`	`}`
`64`	`65`