properly close model

bioimage-io · Oct 3, 2024 · 716a622 · 716a622
1 parent d8c0725
commit 716a622
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java b/src/main/java/io/bioimage/modelrunner/tensorflow/v1/Tensorflow1Interface.java
@@ -514,16 +514,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