include ModalityType

Author: christinadionysio

src/main/python/systemds/scuro/representations/bert.py

@@ -26,6 +26,7 @@
 import torch
 from transformers import BertTokenizer, BertModel
 from systemds.scuro.representations.utils import save_embeddings
+from systemds.scuro.modality.type import ModalityType
 
 
 class Bert(UnimodalRepresentation):

src/main/python/systemds/scuro/representations/bow.py

@@ -25,6 +25,7 @@
 from systemds.scuro.representations.unimodal import UnimodalRepresentation
 from systemds.scuro.representations.utils import save_embeddings
 
+from systemds.scuro.modality.type import ModalityType
 
 class BoW(UnimodalRepresentation):
     def __init__(self, ngram_range=2, min_df=2, output_file=None):

src/main/python/systemds/scuro/representations/glove.py

@@ -24,6 +24,7 @@
 
 from systemds.scuro.representations.unimodal import UnimodalRepresentation
 from systemds.scuro.representations.utils import read_data_from_file, save_embeddings
+from systemds.scuro.modality.type import ModalityType
 
 
 def load_glove_embeddings(file_path):

src/main/python/systemds/scuro/representations/resnet.py

@@ -27,6 +27,7 @@
 import torchvision.models as models
 import torchvision.transforms as transforms
 import numpy as np
+from systemds.scuro.modality.type import ModalityType
 
 if torch.backends.mps.is_available():
     DEVICE = torch.device("mps")

src/main/python/systemds/scuro/representations/tfidf.py

@@ -25,6 +25,7 @@
 from systemds.scuro.representations.unimodal import UnimodalRepresentation
 from systemds.scuro.representations.utils import save_embeddings
 
+from systemds.scuro.modality.type import ModalityType
 
 class TfIdf(UnimodalRepresentation):
     def __init__(self, min_df=2, output_file=None):

src/main/python/systemds/scuro/representations/word2vec.py

@@ -25,6 +25,7 @@
 from gensim.models import Word2Vec
 from gensim.utils import tokenize
 
+from systemds.scuro.modality.type import ModalityType
 import nltk
 
 nltk.download("punkt_tab")