CU-86995mmb6: Remove excessive/unncessary type hints

Author: mart-r

medcat/utils/relation_extraction/llama/model.py

@@ -1,12 +1,11 @@
 import logging
-from typing import Any, Optional, Tuple, Union
+from typing import Any, Optional, Tuple
 import torch
 from torch import nn
 import os
 from transformers.models.llama import LlamaModel
 
 from medcat.config_rel_cat import ConfigRelCAT
-from medcat.utils.relation_extraction.config import BaseConfig_RelationExtraction
 from medcat.utils.relation_extraction.llama.config import LlamaConfig_RelationExtraction
 from medcat.utils.relation_extraction.models import BaseModel_RelationExtraction
 from medcat.utils.relation_extraction.ml_utils import create_dense_layers, get_annotation_schema_tag
@@ -36,7 +35,7 @@ def __init__(self, pretrained_model_name_or_path: str, relcat_config: ConfigRelC
                                                           model_config=model_config)
 
         self.relcat_config: ConfigRelCAT = relcat_config
-        self.model_config: Union[BaseConfig_RelationExtraction, LlamaConfig_RelationExtraction] = model_config
+        self.model_config = model_config
 
         self.hf_model: LlamaModel = LlamaModel(config=model_config.hf_model_config)
 

medcat/utils/relation_extraction/modernbert/model.py

@@ -9,7 +9,6 @@
 from transformers import PreTrainedModel
 from medcat.utils.relation_extraction.ml_utils import create_dense_layers
 from medcat.utils.relation_extraction.models import BaseModel_RelationExtraction
-from medcat.utils.relation_extraction.config import BaseConfig_RelationExtraction
 from medcat.utils.relation_extraction.modernbert.config import ModernBertConfig_RelationExtraction
 
 
@@ -37,7 +36,7 @@ def __init__(self, pretrained_model_name_or_path: str, relcat_config: ConfigRelC
                                                     model_config=model_config)
 
         self.relcat_config: ConfigRelCAT = relcat_config
-        self.model_config: Union[BaseConfig_RelationExtraction, ModernBertConfig_RelationExtraction] = model_config
+        self.model_config = model_config
         self.pretrained_model_name_or_path: str = pretrained_model_name_or_path
 
         self.hf_model: Union[ModernBertModel, PreTrainedModel] = ModernBertModel(config=model_config.hf_model_config)