tweaks

csst_common/pipeline.py

@@ -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}'...")