Prevent LazyLoader loading module when repr is called on module (#…

…10494) Signed-off-by: Weichen Xu <weichen.xu@databricks.com>
mlflow · Nov 27, 2023 · 7dfbd2a · 7dfbd2a
1 parent 34c5bc2
commit 7dfbd2a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mlflow/utils/lazy_load.py b/mlflow/utils/lazy_load.py
@@ -44,3 +44,8 @@ def __getattr__(self, item):
     def __dir__(self):
         module = self._load()
         return dir(module)
+
+    def __repr__(self):
+        if not self._module:
+            return f"<module '{self.__name__} (Not loaded yet)'>"
+        return repr(self._module)