diff --git a/tests/test_data.py b/tests/test_data.py index 813d165a..c67bcc8d 100644 --- a/tests/test_data.py +++ b/tests/test_data.py @@ -38,6 +38,7 @@ class TestLMM: def generate(self, _, **kwargs): return "test" + class TestEmb: def embed(self, _): return np.random.randn(128).astype(np.float32)