Do not enforce unnecessary dependencies

ff74b1b9 · Niklas Muennighoff · GitHub · e81d3cce · ff74b1b9
Unverified Commit ff74b1b9 authored Oct 12, 2023 by Niklas Muennighoff Committed by GitHub Oct 12, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/base.py lm_eval/base.py +1 -1

No files found.
--- a/lm_eval/base.py
+++ b/lm_eval/base.py
@@ -7,7 +7,6 @@ import os
 import json
 import hashlib
 import datasets
-from sqlitedict import SqliteDict
 from tqdm import tqdm
 import torch
 import torch.nn.functional as F
@@ -891,6 +890,7 @@ class CachingLM:
        :param cache_db: str
            Path to cache db
        """
+        from sqlitedict import SqliteDict
        self.lm = lm
        self.cache_db = cache_db
        if os.path.dirname(cache_db):