Initialize the TaskEncoder. Args: checkpoint: Path to the pretrained model. context: Python module context in which tasks are defined. batch_size: Size of each batch during embedding computation. tmp_file_path: Temporary file path for saving intermed
(
self,
checkpoint: str,
context: ModuleType,
batch_size=2,
tmp_file_path="tmp_task_encoder.pkl",
)
source not stored for this graph (policy: none)
nothing calls this directly
no test coverage detected