fix typo

maps-as-data · Oct 31, 2023 · bbb2c8f · bbb2c8f
1 parent d99df36
commit bbb2c8f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_classify/test_classifier.py b/tests/test_classify/test_classifier.py
@@ -195,7 +195,7 @@ def test_init_models_timm(inputs):
         assert isinstance(classifier.model, model_type)
         assert all(k in classifier.dataloaders.keys() for k in ["train", "test", "val"])
         classifier = ClassifierContainer(
-            model, labels_map=annots.labels_map
+            my_model, labels_map=annots.labels_map
         )
         assert isinstance(classifier.model, model_type)
         assert classifier.dataloaders == {}