naming fixes

nabirds/annotations/__init__.py

@@ -5,8 +5,8 @@ from .cars import CARS_Annotations
 from cvargparse.utils import BaseChoiceType
 
 class AnnotationType(BaseChoiceType):
-	CUB = CUB_Annotations
+	CUB200 = CUB_Annotations
 	NAB = NAB_Annotations
 	CARS = CARS_Annotations
 
-	Default = CUB
+	Default = CUB200

nabirds/annotations/base.py

@@ -87,7 +87,7 @@ class BaseAnnotations(abc.ABC):
 
 		if None not in [subset, self.feature_model]:
 			tried = []
-			model_info = self.info[self.feature_model]
+			model_info = self.info.MODELS[self.feature_model]
 			for subset_phony in BaseAnnotations.FEATURE_PHONY[subset]:
 				features = feature_file_name(subset_phony, dataset_info, model_info)
 				feature_path = join(self.root, "features", features)