[FIX] Make input_size primitive type

wbenbihi · Aug 20, 2022 · d9bb429 · d9bb429
1 parent 00b1c7e
commit d9bb429
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/config/config.default.yaml b/config/config.default.yaml
@@ -49,13 +49,12 @@ data:
 
 dataset:
   object: "hourglass_tensorflow.datasets.HTFDatasetHandler"
+  input_size: 256
   params:
     mockup_param: to_pass_to_object_init_method
   sets:
     test: true
     validation: true
-  input:
-    size: 256
   split:
     column: set
     train_value: TRAIN

diff --git a/hourglass_tensorflow/utils/config.py b/hourglass_tensorflow/utils/config.py
@@ -68,17 +68,13 @@ class HTFDatasetBBoxConfig(BaseModel):
     )
 
 
-class HTFDatasetSizeConfig(BaseModel):
-    input: int = 256
-
-
 class HTFDatasetConfig(BaseModel):
     object: str
+    input_size: int = 256
     params: Optional[HTFDatasetParamsConfig] = Field(
         default_factory=HTFDatasetParamsConfig
     )
     sets: Optional[HTFDatasetSetsConfig]
-    size: Optional[HTFDatasetSizeConfig] = Field(default_factory=HTFDatasetSizeConfig)
     split: Optional[HTFDatasetSplitConfig] = Field(
         default_factory=HTFDatasetSplitConfig
     )