fix shorttext encoder is_trainable

mindsdb · Jun 13, 2023 · ac3ee69 · ac3ee69
1 parent 9636253
commit ac3ee69
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lightwood/api/json_ai.py b/lightwood/api/json_ai.py
@@ -173,8 +173,7 @@ def lookup_encoder(
     if encoder_dict["module"] == "PretrainedLangEncoder" and not is_target:
         encoder_dict["args"]["output_type"] = "$dtype_dict[$target]"
 
-    enc_cls = eval(encoder_dict["module"])
-    if enc_cls.is_trainable_encoder and hasattr(enc_cls, 'stop_after'):
+    if eval(encoder_dict["module"]).is_trainable_encoder:
         encoder_dict["args"]["stop_after"] = "$problem_definition.seconds_per_encoder"
 
     if is_target_predicting_encoder:

diff --git a/lightwood/encoder/text/short.py b/lightwood/encoder/text/short.py
@@ -8,6 +8,8 @@
 
 
 class ShortTextEncoder(BaseEncoder):
+    is_trainable_encoder = False
+
     def __init__(self, is_target=False, mode=None, device=''):
         """
         :param is_target: