📝 Add comment explaining wip comment for prompt tuning

Signed-off-by: gkumbhat <[email protected]>
Signed-off-by: Sukriti-Sharma4 <[email protected]>

Author: gkumbhat

caikit_nlp/modules/text_generation/peft_prompt_tuning.py

@@ -49,7 +49,7 @@
 from caikit import get_config
 from caikit.core.data_model import DataStream
 from caikit.core.modules import ModuleBase, ModuleConfig, ModuleSaver, module
-from caikit.core.toolkit import error_handler, wip_decorator
+from caikit.core.toolkit import error_handler
 from caikit.interfaces.nlp.data_model import (
     ClassificationTrainRecord,
     GeneratedTextResult,
@@ -213,6 +213,9 @@ def run(
             )
         return GeneratedTextResult(generated_text=gen_text[0])
 
+    # NOTE: We need to disable wip decorator here otherwise we get issues in
+    # proto generation for streaming. We are keeping it commented out for now,
+    # to essentially document that this streaming function is WIP.
     # @wip_decorator.work_in_progress(
     #     category=wip_decorator.WipCategory.WIP, action=wip_decorator.Action.WARNING
     # )