Minor changes in config

hezarai · Aug 20, 2023 · 5501d65 · 5501d65
1 parent b5a69f5
commit 5501d65
Showing 1 changed file with 1 addition and 14 deletions.
diff --git a/hezar/configs.py b/hezar/configs.py
@@ -41,19 +41,6 @@
 
 logger = get_logger(__name__)
 
-CONFIG_TYPES_MAPPING = {
-    "Config": ConfigType.BASE,
-    "ModelConfig": ConfigType.MODEL,
-    "PreprocessorConfig": ConfigType.PREPROCESSOR,
-    "TrainerConfig": ConfigType.TRAINER,
-    "DatasetConfig": ConfigType.DATASET,
-    "EmbeddingConfig": ConfigType.EMBEDDING,
-    "CriterionConfig": ConfigType.CRITERION,
-    "OptimizerConfig": ConfigType.OPTIMIZER,
-    "LRSchedulerConfig": ConfigType.LR_SCHEDULER,
-    "MetricConfig": ConfigType.METRIC,
-}
-
 CONFIG_CLASS_VARS = ["name", "config_type"]
 
 
@@ -73,7 +60,7 @@ def __getitem__(self, item):
         try:
             return self.dict()[item]
         except KeyError:
-            raise AttributeError(f"`{self.__class__.__name__}` has no attribute `{item}`!")
+            raise AttributeError(f"`{self.__class__.__name__}` does not have the parameter `{item}`!")
 
     def __len__(self):
         return len(self.dict())