fix: fix broken tests

Signed-off-by: eshwarprasadS <eshwarprasad.s01@gmail.com>

Author: eshwarprasadS

src/instructlab/sdg/encoders/arctic_encoder.py

@@ -116,13 +116,13 @@ def _initialize_model(self) -> None:
                     "Please run `ilab model download` and download the necessary model"
                 )
 
-        self.tokenizer = AutoTokenizer.from_pretrained(model_path)
-        self.model = AutoModel.from_pretrained(
-            model_path,
-            add_pooling_layer=False,
-            trust_remote_code=True,
-            local_files_only=True,
-        )
+            self.tokenizer = AutoTokenizer.from_pretrained(model_path)
+            self.model = AutoModel.from_pretrained(
+                model_path,
+                add_pooling_layer=False,
+                trust_remote_code=True,
+                local_files_only=True,
+            )
 
         if self.cfg.use_fp16:
             self.model = self.model.half()

tests/test_subset_selection.py

@@ -33,10 +33,11 @@ def mock_gpu_environment():
 @pytest.fixture
 def mock_encoder():
     class MockEncoder:
-        def __init__(self, model_name):
+        def __init__(self, model_name, testing_mode=False):
             self.model_name = model_name
+            self.testing_mode = testing_mode
 
-        def encode(self, inputs, instruction, query_description):
+        def encode(self, inputs, instruction):
             # Return mock embeddings
             return torch.randn(len(inputs), 768)