From d48815cca29951705cbd43cc7599b14c6b3c103d Mon Sep 17 00:00:00 2001 From: GassiGiuseppe Date: Sun, 5 Oct 2025 18:58:42 +0200 Subject: [PATCH] added task_type and updated init --- Project_Model/Libs/Transformer/Utils/__init__.py | 3 ++- Project_Model/Libs/Transformer/Utils/task_type.py | 6 ++++++ 2 files changed, 8 insertions(+), 1 deletion(-) create mode 100644 Project_Model/Libs/Transformer/Utils/task_type.py diff --git a/Project_Model/Libs/Transformer/Utils/__init__.py b/Project_Model/Libs/Transformer/Utils/__init__.py index 856b51f..2831ec4 100644 --- a/Project_Model/Libs/Transformer/Utils/__init__.py +++ b/Project_Model/Libs/Transformer/Utils/__init__.py @@ -1,3 +1,4 @@ from .attention_mask import get_attention_mask +from .task_type import TaskType -__all__ = ["get_attention_mask"] \ No newline at end of file +__all__ = ["get_attention_mask", "TaskType"] \ No newline at end of file diff --git a/Project_Model/Libs/Transformer/Utils/task_type.py b/Project_Model/Libs/Transformer/Utils/task_type.py new file mode 100644 index 0000000..46a42eb --- /dev/null +++ b/Project_Model/Libs/Transformer/Utils/task_type.py @@ -0,0 +1,6 @@ +from enum import Enum, auto + +class TaskType(Enum): + RDF2TEXT = auto() + MASK = auto() + COMPLETATION = auto() \ No newline at end of file