dataset type integration contd.

aidos-lab · Aug 31, 2024 · ae3ef4e · ae3ef4e
1 parent 71211b7
commit ae3ef4e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/code/datasets/transforms.py b/code/datasets/transforms.py
@@ -48,7 +48,6 @@ def __call__(self, data):
         data.face = torch.tensor(data.triangulation).T - 1
         data.num_nodes = data.face.max() + 1
         data.triangulation = None
-        print(data, "m")
         return data
 
 

diff --git a/code/experiments/utils/run_experiment.py b/code/experiments/utils/run_experiment.py
@@ -31,7 +31,9 @@ def get_setup(
 ) -> Tuple[SimplicialDataModule, BaseModel, L.Trainer, WandbLogger]:
     run_id = str(uuid.uuid4())
     transforms = transforms_lookup[config.transforms]
-    task_lookup: Dict[TaskType, Task] = get_task_lookup(transforms)
+    task_lookup: Dict[TaskType, Task] = get_task_lookup(
+        transforms, ds_type=config.ds_type
+    )
 
     dm = SimplicialDataModule(
         ds_type=config.ds_type,