Commit 2559b176 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

tweaks

parent 23c06579

csst_common/pipeline.py

+9 −2
Original line number Diff line number Diff line
@@ -31,7 +31,7 @@ class Pipeline: 


        self.n_jobs_cpu = n_jobs_cpu

        self.n_jobs_gpu = n_jobs_gpu

        self.device = device

        self.device = device.upper()



        self.pipeline_logger = get_logger(name="pipeline", filename=os.path.join(self.dir_out, "pipeline.log"))

        self.module_logger = get_logger(name="module", filename=os.path.join(self.dir_out, "module.log"))
@@ -45,10 +45,17 @@ class Pipeline: 
            self.clean_directory(self.dir_out)



        # Frequently used files

        self.msg_file = Message(os.path.join(self.dir_out, "msg.txt"))

        self.message = Message(os.path.join(self.dir_out, "msg.txt"))

        self.time_stamp = TimeStamp(os.path.join(self.dir_out, "time_stamp.txt"))

        self.exit_code = ExitCode(os.path.join(self.dir_out, "exit_code"))



    def set_test_env(self):

        os.environ["CSST_DFS_API_MODE"] = "cluster"

        os.environ["CSST_DFS_GATEWAY"] = "172.24.27.2:30880"

        os.environ["CSST_DFS_APP_ID"] = "test"

        os.environ["CSST_DFS_APP_TOKEN"] = "test"

        os.environ["CRDS_SERVER_URL"] = "http://172.24.27.2:29000"



    @staticmethod

    def clean_directory(d):

        print(f"Clean output directory '{d}'...")