]> mj.ucw.cz Git - eval.git/blobdiff - t/moe/eval.py
Renamed conf -> config, refactoring inside config
[eval.git] / t / moe / eval.py
index 3e313f34e47032aa4d07cb00ab3411b6c7137839..2f30c6d2e1148a67b33bfc69c84badd2af79293a 100644 (file)
@@ -2,8 +2,9 @@
 
 import moe
 import moe.config
+import moe.box
 import moe.log
-import moe.meta
+import moe.status
 import moe.pipeline
 import moe.batch
 import moe.util
@@ -18,64 +19,69 @@ class Eval:
        self.builtins = moe.config.MoeConfig(type="builtins")
        self.cfgs.push(self.builtins)
        self.main_pipe = moe.pipeline.MoePipeline("main")
-       self.meta = moe.meta.MoeMeta()
+       self.test_pipe = moe.pipeline.MoePipeline("test")
+       self.stat = moe.status.MoeStatus()
        pass
 
+    def __getitem__(self, key):
+       return self.cfgs[key]
+
     def init(self, overrides=None):
        self.log.progress("Initializing... ")
        self.init_global(overrides)
        self.init_test()
        self.init_logs()
        self.init_task()
+       moe.box.init(self)
        self.log.progress("OK\n")
 
     def init_global(self, overrides):
-       main_cfg = moe.config.MoeConfig(name = os.path.join(self.cfgs['HOME'], "config"), type="main")
+       main_cfg = moe.config.MoeConfig(name = os.path.join(self['HOME'], "config"), type="main")
        self.cfgs.push(main_cfg)
        if overrides:
            self.cfgs.push(overrides)
 
     def init_test(self):
-       test = self.cfgs['TEST_DIR']
+       test = self['TDIR']
        if os.path.isdir(test):
            shutil.rmtree(test)
        try:
            moe.util.mkdir_tree(test)   
-       except OSError, e:
-           raise moe.MoeErr, "Cannot create %s: %s" % (test, e.strerror)
+       except OSError, err:
+           raise moe.MoeError, "Cannot create %s: %s" % (test, err.strerror)
 
     def init_logs(self):
        self.log = moe.log.MoeLog()
-       if self.cfgs["V"]:
-           self.log.verbosity = int(self.cfgs["V"])
-       self.log.log_file = open(os.path.join(self.cfgs["TEST_DIR"], "log"), "w")
+       if self["V"]:
+           self.log.verbosity = int(self["V"])
+       self.log.open(os.path.join(self["TDIR"], "log"))
        self.default_log = self.log
        moe.log.default = self.log
        self.log_config(3, "before loading the task")
 
     def init_task(self):
-       task = self.cfgs['TASK']
-       task_dir = self.cfgs['TASK_DIR']
+       task = self['TASK']
+       task_dir = self['PDIR']
        if not os.path.isdir(task_dir):
-           raise moe.MoeErr, "No such task %s" % task
+           raise moe.MoeError, "No such task %s" % task
 
        task_cfg = moe.config.MoeConfig(name = os.path.join(task_dir, "config"), type='task')
        self.cfgs.push(task_cfg)
        self.log_config(3, "after loading the task")
 
-       self.meta["task"] = task
+       self.stat["task"] = task
 
-       type = self.cfgs['TASK_TYPE']
+       type = self['TASK_TYPE']
        if type == "batch" or type == "interactive":
            moe.batch.prepare_pipe(self)
        elif type == "opendata":
-           raise moe.MoeErr, "Opendata tasks not implemented yet"
+           raise moe.MoeError, "Opendata tasks not implemented yet"
        else:
-           raise moe.MoeErr, "Unknown task type " + type
+           raise moe.MoeError, "Unknown task type " + type
 
     def run(self):
        self.log_config(2, "for the task pipeline")
-       self.main_pipe.configure(self.cfgs["HOOKS"])
+       self.main_pipe.configure(self["HOOKS"])
        if self.log.verbosity >= 2:
            self.main_pipe.dump(self.log.log_file, prefix="\t")
        self.main_pipe.run(self)