Fix test

Eventual-Inc · Sep 11, 2024 · d9cd2d6 · d9cd2d6
1 parent d294846
commit d9cd2d6
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/daft/lazy_import.py b/daft/lazy_import.py
@@ -25,16 +25,14 @@ class LazyImport:
     """
 
     def __init__(self, module_name: str):
-        self._import_attempted = False
         self._module_name = module_name
         self._module = None
 
     def module_available(self):
-        return self._load_module is not None
+        return self._load_module() is not None
 
     def _load_module(self):
-        if not self._import_attempted and self._module is None:
-            self._import_attempted = True
+        if self._module is None:
             try:
                 self._module = importlib.import_module(self._module_name)
             except ImportError: