Method _is_base_py_module_context

python/tvm/script/parser/core/parser.py:469–477

Check if the current class context allows Python functions. Returns ------- bool True if Python functions are allowed in the current context.

(self)

        return self.current_class

    def _is_base_py_module_context(self) -> bool:
        """Check if the current class context allows Python functions.

        Returns
        -------
        bool
            True if Python functions are allowed in the current context.
        """
        return self.base_py_module_context

    def with_diag_source(self, source: Source):
        """Add a new source as with statement.