properly close model

carlosuc3m · carlosuc3m · commit eabb9ac51eb6 · 2024-10-03T13:41:34.000+02:00
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api020/Tensorflow2Interface.java
@@ -502,16 +502,17 @@ else if (task.status == TaskStatus.FAILED)
 			else if (task.status == TaskStatus.CRASHED)
 				throw new RuntimeException();
 			this.runner.close();
+			this.runner = null;
 			return;
 		} else if (this.interprocessing) {
 			return;
 		}
-		sig = null;
 		if (model != null) {
 			model.session().close();
 			model.close();
 		}
 		model = null;
+		sig = null;
 	}
 
 	// TODO make only one

Original file line number	Diff line number	Diff line change
`@@ -502,16 +502,17 @@ else if (task.status == TaskStatus.FAILED)`
`502`	`502`	`else if (task.status == TaskStatus.CRASHED)`
`503`	`503`	`throw new RuntimeException();`
`504`	`504`	`this.runner.close();`
	`505`	`+ this.runner = null;`
`505`	`506`	`return;`
`506`	`507`	`} else if (this.interprocessing) {`
`507`	`508`	`return;`
`508`	`509`	`}`
`509`		`- sig = null;`
`510`	`510`	`if (model != null) {`
`511`	`511`	`model.session().close();`
`512`	`512`	`model.close();`
`513`	`513`	`}`
`514`	`514`	`model = null;`
	`515`	`+ sig = null;`
`515`	`516`	`}`
`516`	`517`
`517`	`518`	`// TODO make only one`