Removed old mantra dependency

.gitignore

@@ -171,4 +171,6 @@ cython_debug/
 #  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-#.idea/
+#.idea/
+
+/bu_mantra

mantra/dataset.py → datasets/simplicial.py

@@ -6,7 +6,6 @@
 import torch
 from torch.utils.data import Subset
 import numpy as np
-from numpy.random import RandomState
 
 
 class SimplicialDataModule(LightningDataModule):

mantra/transforms.py → datasets/transforms.py

@@ -8,10 +8,10 @@
 )
 from torch_geometric.transforms import ToUndirected
 from torch_geometric.utils import degree
-import torchvision.transforms as transforms
+
+# import torchvision.transforms as transforms
 from torch_geometric.transforms import FaceToEdge, OneHotDegree
-import mantra.k_simplex2vec as ks2v
-from mantra.utils import (
+from datasets.utils import (
     create_signals_on_data_if_needed,
     append_signals,
     create_other_features_on_data_if_needed,

experiments/configs.py

@@ -4,7 +4,7 @@
 
 from pydantic_settings import BaseSettings
 from models.models import ModelConfig
-from mantra import TransformType
+from datasets.transforms import TransformType
 from metrics.tasks import TaskType
 from models.models import ModelType
 import yaml

experiments/run_experiment.py

@@ -10,12 +10,12 @@
 from experiments.configs import ConfigExperimentRun
 from models import model_lookup
 from typing import Dict
-from mantra.dataset import SimplicialDataModule
+from datasets.simplicial import SimplicialDataModule
 from models.base import BaseModel
 from experiments.loggers import get_wandb_logger
 import lightning as L
 import uuid
-from mantra import transforms_lookup
+from datasets.transforms import transforms_lookup
 
 
 def run_configuration(config: ConfigExperimentRun):

generate_configs.py

@@ -1,5 +1,5 @@
 from metrics.tasks import TaskType
-from mantra.transforms import TransformType
+from datasets.transforms import TransformType
 from models import ModelType, model_cfg_lookup
 from experiments.configs import ConfigExperimentRun, TrainerConfig
 import yaml