Fixing a small issue in OpRegularizerManager.

Internally, OpRegularizerManager was passing output_boundary as a list to _dfs_for_source_ops, which was packaging it again as a list. The fix now allows _dfs_for_source_ops to support a single output_boundary node or a list of nodes.

PiperOrigin-RevId: 300625458

Author: mn-robot

morph_net/framework/op_regularizer_manager.py

@@ -599,7 +599,10 @@ def _dfs_for_source_ops(self, output_boundary, input_boundary=None):
       input_boundary = set(input_boundary)
     else:
       input_boundary = set()
-    to_visit = list(output_boundary)
+    if isinstance(output_boundary, list):
+      to_visit = output_boundary
+    else:
+      to_visit = list(output_boundary)
     visited = set()
     while to_visit:
       # Get next op and mark as visited.