properly close model

carlosuc3m · carlosuc3m · commit 65166378e369 · 2024-10-03T13:41:19.000+02:00
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api030/Tensorflow2Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v2/api030/Tensorflow2Interface.java
@@ -489,16 +489,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
`@@ -489,16 +489,17 @@ else if (task.status == TaskStatus.FAILED)`
`489`	`489`	`else if (task.status == TaskStatus.CRASHED)`
`490`	`490`	`throw new RuntimeException();`
`491`	`491`	`this.runner.close();`
	`492`	`+ this.runner = null;`
`492`	`493`	`return;`
`493`	`494`	`} else if (this.interprocessing) {`
`494`	`495`	`return;`
`495`	`496`	`}`
`496`		`- sig = null;`
`497`	`497`	`if (model != null) {`
`498`	`498`	`model.session().close();`
`499`	`499`	`model.close();`
`500`	`500`	`}`
`501`	`501`	`model = null;`
	`502`	`+ sig = null;`
`502`	`503`	`}`
`503`	`504`
`504`	`505`	`// TODO make only one`