force remove ref output plug datum on unplug

samuelmtimbo · Mar 1, 2024 · b77be77 · b77be77
1 parent feb1a2a
commit b77be77
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/Class/Graph/index.ts b/src/Class/Graph/index.ts
@@ -2166,15 +2166,17 @@ export class Graph<I = any, O = any>
         propagate
       )
 
-      if (propagate) {
-        if (isUnitPinRef) {
-          if (isOutput) {
-            pin.take()
-          } else {
+      if (isUnitPinRef) {
+        if (isOutput) {
+          pin.take()
+        } else {
+          if (propagate) {
             unit.takePin(type, _pinId)
           }
         }
+      }
 
+      if (propagate) {
         if (type === 'input') {
           unit.takeInput(_pinId)
         }