소스 검색

small change

Dimitri Korsch 5 년 전
부모
커밋
c7d12fea69
1개의 변경된 파일1개의 추가작업 그리고 4개의 파일을 삭제
  1. 1 4
      cvdatasets/annotation/base.py

+ 1 - 4
cvdatasets/annotation/base.py

@@ -49,10 +49,7 @@ class BaseAnnotations(abc.ABC):
 	@property
 	@only_with_info
 	def dataset_info(self):
-		key = getattr(self.__class__, "name", None)
-
-		if key is None:
-			key = self.dataset_key
+		key = getattr(self.__class__, "name", self.dataset_key)
 
 		if key not in self.info.DATASETS:
 			raise ValueError(f"Cannot find dataset with key \"{key}\"")