Update dfa.py

caleb531 · Jun 18, 2023 · e51fb59 · e51fb59
1 parent 41fc43c
commit e51fb59
Showing 1 changed file with 11 additions and 17 deletions.
diff --git a/automata/fa/dfa.py b/automata/fa/dfa.py
@@ -1865,30 +1865,24 @@ def get_name_original(states: FrozenSet[DFAStateT]) -> DFAStateT:
                     dfa_states.add(next_current_states_name)
                     state_queue.append(next_current_states)
 
-        # if minify:
-        #    return cls._minify(
-        #        reachable_states=dfa_states,
-        #        input_symbols=dfa_symbols,
-        #        transitions=dfa_transitions,
-        #        initial_state=dfa_initial_state,
-        #        reachable_final_states=dfa_final_states,
-        #        retain_names=retain_names,
-        #    )
-        # TODO change this back once _minify has compatibility with partial DFAs.
-
-        final_dfa = cls._to_complete(
+        if minify:
+            return cls._minify(
+                reachable_states=dfa_states,
+                input_symbols=dfa_symbols,
+                transitions=dfa_transitions,
+                initial_state=dfa_initial_state,
+                reachable_final_states=dfa_final_states,
+                retain_names=retain_names,
+            )
+
+        return cls._to_complete(
             input_symbols=dfa_symbols,
             transitions=dfa_transitions,
             initial_state=dfa_initial_state,
             final_states=dfa_final_states,
             trap_state=frozenset(),
         )
 
-        if minify:
-            final_dfa = final_dfa.minify(retain_names=retain_names)
-
-        return final_dfa
-
     def iter_transitions(
         self,
     ) -> Generator[Tuple[DFAStateT, DFAStateT, str], None, None]: