Update project_to_rewrites

runtimeverification · Dec 18, 2024 · e83fa6f · e83fa6f
1 parent 928f855
commit e83fa6f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pyk/src/pyk/kore/internal.py b/pyk/src/pyk/kore/internal.py
@@ -103,10 +103,12 @@ def project_to_symbols(self) -> KoreDefn:
         )
 
     def project_to_rewrites(self) -> KoreDefn:
-        """Project definition to symbols that are (transitively) referred to from rewrite axioms."""
-        rewrite_symbols = self._rewrite_symbols()
+        """Project definition to symbols that are part of the configuration or are (transitively) referred to from rewrite axioms."""
+        _symbols = set()
+        _symbols.update(self._config_symbols())
+        _symbols.update(self._rewrite_symbols())
         symbols: FrozenDict[str, SymbolDecl] = FrozenDict(
-            (symbol, decl) for symbol, decl in self.symbols.items() if symbol in rewrite_symbols
+            (symbol, decl) for symbol, decl in self.symbols.items() if symbol in _symbols
         )
         return self.let(symbols=symbols).project_to_symbols()