properly close model

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