From 410f9dbfc8c5429d981b96664dac0e039b9555e2 Mon Sep 17 00:00:00 2001 From: Benjamin Fineran Date: Wed, 26 Jan 2022 17:40:58 -0500 Subject: [PATCH] add flag to signal NM integration is active (#32) --- src/transformers/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index 26be362c5dab..b80633386607 100755 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -2779,6 +2779,9 @@ class _LazyModule(_BaseLazyModule): __file__ = globals()["__file__"] __path__ = [os.path.dirname(__file__)] + # flag to signal NM integration is active + NM_INTEGRATED = True + def _get_module(self, module_name: str): return importlib.import_module("." + module_name, self.__name__)