update test for cli change

Author: aditya0by0

tests/unit/cli/classification_labels.txt

@@ -0,0 +1,10 @@
+label_1
+label_2
+label_3
+label_4
+label_5
+label_6
+label_7
+label_8
+label_9
+label_10

tests/unit/cli/mock_dm.py

@@ -1,3 +1,5 @@
+import os
+
 import torch
 from lightning.pytorch.core.datamodule import LightningDataModule
 from torch.utils.data import DataLoader
@@ -29,6 +31,10 @@ def num_of_labels(self):
     def feature_vector_size(self):
         return self._feature_vector_size
 
+    @property
+    def classes_txt_file_path(self) -> str:
+        return os.path.join("tests", "unit", "cli", "classification_labels.txt")
+
     def train_dataloader(self):
         assert self.feature_vector_size is not None, "feature_vector_size must be set"
         # Dummy dataset for example purposes

tests/unit/cli/testCLI.py

@@ -9,7 +9,7 @@ def setUp(self):
             "fit",
             "--trainer=configs/training/default_trainer.yml",
             "--model=configs/model/ffn.yml",
-            "--model.init_args.hidden_layers=[10]",
+            "--model.init_args.hidden_layers=[1]",
             "--model.train_metrics=configs/metrics/micro-macro-f1.yml",
             "--data=tests/unit/cli/mock_dm_config.yml",
             "--model.pass_loss_kwargs=false",