Documentation and formatting updates #1
|
@ -6,11 +6,6 @@ from typing import Dict
|
||||||
import yaml
|
import yaml
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class GeneralConfig:
|
|
||||||
run_mode: str
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class DataSetConfig:
|
class DataSetConfig:
|
||||||
num_slices: int
|
num_slices: int
|
||||||
|
@ -54,7 +49,6 @@ class AppSettings:
|
||||||
config_data = yaml.safe_load(f)
|
config_data = yaml.safe_load(f)
|
||||||
|
|
||||||
# Parse the loaded YAML into dataclass objects
|
# Parse the loaded YAML into dataclass objects
|
||||||
self.general = GeneralConfig(**config_data["general"])
|
|
||||||
self.dataset = DataSetConfig(**config_data["dataset"])
|
self.dataset = DataSetConfig(**config_data["dataset"])
|
||||||
self.training = TrainingConfig(**config_data["training"])
|
self.training = TrainingConfig(**config_data["training"])
|
||||||
self.app = AppConfig(**config_data["app"])
|
self.app = AppConfig(**config_data["app"])
|
||||||
|
|
Loading…
Reference in New Issue
Block a user