pyduino.paths

View Source

 1import os
 2import yaml
 3import warnings
 4
 5__location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
 6
 7def yaml_get(filename):
 8    """
 9    Loads hyperparameters from a YAML file.
10    """
11    y = None
12    with open(filename) as f:
13        y = yaml.load(f.read(),yaml.Loader)
14    return y
15
16class Paths():
17    def read(self,config_path):
18        self.CONFIG_PATH = config_path
19        self.SYSTEM_PARAMETERS = yaml_get(self.CONFIG_PATH)['system']
20        self.SLAVE_PARAMETERS = yaml_get(self.CONFIG_PATH)['slave']
21        self.RELEVANT_PARAMETERS = self.SYSTEM_PARAMETERS['relevant_parameters']
22        self.HYPERPARAMETERS = yaml_get(self.CONFIG_PATH)['hyperparameters']
23        self.INITIAL_STATE_PATH = os.path.join(__location__,self.SYSTEM_PARAMETERS['initial_state'])
24        self.SCHEMA = self.SYSTEM_PARAMETERS["standard_parameters"]
25        self.REACTOR_PARAMETERS = list(self.SCHEMA.keys())
26        self.TENSORBOARD = self.SYSTEM_PARAMETERS.get("tensorboard", None)
27
28PATHS = Paths()
29if os.path.exists(os.path.join(__location__,"config.yaml")):
30    PATHS.read(os.path.join(__location__,"config.yaml"))
31else:
32    warnings.warn("No config.yaml file found in the current directory. Please create one.")
33    PATHS.read(os.path.join(__location__,"config.template.yaml"))

def yaml_get(filename): View Source

 8def yaml_get(filename):
 9    """
10    Loads hyperparameters from a YAML file.
11    """
12    y = None
13    with open(filename) as f:
14        y = yaml.load(f.read(),yaml.Loader)
15    return y

Loads hyperparameters from a YAML file.

class Paths: View Source

17class Paths():
18    def read(self,config_path):
19        self.CONFIG_PATH = config_path
20        self.SYSTEM_PARAMETERS = yaml_get(self.CONFIG_PATH)['system']
21        self.SLAVE_PARAMETERS = yaml_get(self.CONFIG_PATH)['slave']
22        self.RELEVANT_PARAMETERS = self.SYSTEM_PARAMETERS['relevant_parameters']
23        self.HYPERPARAMETERS = yaml_get(self.CONFIG_PATH)['hyperparameters']
24        self.INITIAL_STATE_PATH = os.path.join(__location__,self.SYSTEM_PARAMETERS['initial_state'])
25        self.SCHEMA = self.SYSTEM_PARAMETERS["standard_parameters"]
26        self.REACTOR_PARAMETERS = list(self.SCHEMA.keys())
27        self.TENSORBOARD = self.SYSTEM_PARAMETERS.get("tensorboard", None)

def read(self, config_path): View Source

18    def read(self,config_path):
19        self.CONFIG_PATH = config_path
20        self.SYSTEM_PARAMETERS = yaml_get(self.CONFIG_PATH)['system']
21        self.SLAVE_PARAMETERS = yaml_get(self.CONFIG_PATH)['slave']
22        self.RELEVANT_PARAMETERS = self.SYSTEM_PARAMETERS['relevant_parameters']
23        self.HYPERPARAMETERS = yaml_get(self.CONFIG_PATH)['hyperparameters']
24        self.INITIAL_STATE_PATH = os.path.join(__location__,self.SYSTEM_PARAMETERS['initial_state'])
25        self.SCHEMA = self.SYSTEM_PARAMETERS["standard_parameters"]
26        self.REACTOR_PARAMETERS = list(self.SCHEMA.keys())
27        self.TENSORBOARD = self.SYSTEM_PARAMETERS.get("tensorboard", None)

PATHS = <Paths object>