Skip to content

Commit 6001597

Browse files
committed
tabR lazy imports
1 parent 74088a7 commit 6001597

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

mambular/base_models/tabr.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def _lazy_import_dependencies(self):
146146

147147
except ImportError:
148148
raise ImportError("Failed to import delu module for TabR. Ensure all dependencies are installed\n"
149-
"You can install faiss running 'pip install delu'.") from None
149+
"You can install delu running 'pip install delu'.") from None
150150

151151
if TabR.faiss is None:
152152
try:
@@ -158,8 +158,8 @@ def _lazy_import_dependencies(self):
158158
print("Successfully lazy imported faiss dependency")
159159

160160
except ImportError as e:
161-
raise ImportError("Failed to import a required module for TabR. Ensure all dependencies are installed\n"
162-
"You can install delu by running 'pip install delu'.") from None
161+
raise ImportError("Failed to import faiss module for TabR. Ensure all dependencies are installed\n"
162+
"You can install faiss running 'pip install faiss-cpu' for CPU and 'pip install faiss-gpu' for GPU.") from None
163163

164164
def _encode(
165165
self,

0 commit comments

Comments
 (0)