fix test

a-r-j · Dec 26, 2023 · b15bdf2 · b15bdf2
1 parent 8d32aa2
commit b15bdf2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/task/test_instantiate_transforms.py b/tests/task/test_instantiate_transforms.py
@@ -15,7 +15,7 @@ def test_instantiate_transforms():
         cfg = omegaconf.OmegaConf.load(config_path)
 
         if t == "multihot_label_encoding.yaml":
-            cfg.num_classes = 2
+            cfg.multihot_label_encoding.num_classes = 2
         transform = instantiate(cfg)
 
         if t == "none.yaml":
@@ -30,7 +30,7 @@ def test_transform_call(example_batch):
         cfg = omegaconf.OmegaConf.load(config_path)
 
         if t == "multihot_label_encoding.yaml":
-            cfg.num_classes = 2
+            cfg.multihot_label_encoding.num_classes = 2
 
         transform = instantiate(cfg)