feat: use property instead of setter

modeling_lora.py

@@ -186,7 +186,7 @@ class BertLoRA(BertPreTrainedModel):
         for name, param in super().named_parameters():
             if "lora" not in name:
                 param.requires_grad_(False)
-        self.select_task(0)
+        self.current_task = 0
 
     @classmethod
     def from_bert(cls, *args, num_adaptions=1, **kwargs):
@@ -194,7 +194,6 @@ class BertLoRA(BertPreTrainedModel):
         config = JinaBertConfig.from_pretrained(*args, **kwargs)
         return cls(config, bert=bert, num_adaptions=num_adaptions)
 
-
     @classmethod
     def from_pretrained(
         cls,
@@ -213,7 +212,6 @@ class BertLoRA(BertPreTrainedModel):
         # TODO: choose between from_bert and super().from_pretrained
         return cls.from_bert(pretrained_model_name_or_path)
 
-
     def _register_lora(self, num_adaptions=1, rank=4, lora_dropout_p=0.0, lora_alpha=1):
         self.apply(
             partial(
@@ -225,7 +223,13 @@ class BertLoRA(BertPreTrainedModel):
             )
         )
 
-    def select_task(self, task_idx: Union[None, int]):
+    @property
+    def current_task(self):
+        return self._task_idx
+
+    @current_task.setter
+    def current_task(self, task_idx: Union[None, int]):
+        self._task_idx = task_idx
         self.apply(
             partial(LoRAParametrization.select_task_for_layer, task_idx=task_idx)
         )