From dcd8daf79ccac800d6b8e50cc7e9b05a2d971310 Mon Sep 17 00:00:00 2001 From: tristenallen Date: Mon, 4 Mar 2024 10:56:44 -0800 Subject: [PATCH] Make `Loader._process_module` public. As described. Allows pytype drivers to more freely add ASTs to a persistent `Loader` object, to be used in situations where in-place modification of the loaded modules is desired. #pytype-in-cider PiperOrigin-RevId: 612509207 --- pytype/load_pytd.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytype/load_pytd.py b/pytype/load_pytd.py index 0af6aa9ea..2372c5718 100644 --- a/pytype/load_pytd.py +++ b/pytype/load_pytd.py @@ -404,9 +404,9 @@ def load_module(self, mod_info, mod_ast=None): return existing if not mod_ast: mod_ast = self._module_loader.load_ast(mod_info) - return self._process_module(mod_info, mod_ast) + return self.process_module(mod_info, mod_ast) - def _process_module(self, mod_info, mod_ast): + def process_module(self, mod_info, mod_ast): """Create a module from a loaded ast and save it to the loader cache. Args: