Commit c4938e08 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

auto identify node name

parent 19dc7050
...@@ -306,13 +306,11 @@ class CsstMbiDataManager: ...@@ -306,13 +306,11 @@ class CsstMbiDataManager:
return dm return dm
@staticmethod @staticmethod
def quick_dataset(hostname="dandelion", ver_sim="C5.2", dir_l1=".", exposure_id=100): def quick_dataset(ver_sim="C5.2", dir_l1=".", exposure_id=100):
""" quick dataset generator """ quick dataset generator for tests on dandelion or PMO
Parameters Parameters
---------- ----------
hostname:
{dandelion, ubuntu}
ver_sim: ver_sim:
{"C5.2"} {"C5.2"}
dir_l1: dir_l1:
...@@ -325,6 +323,9 @@ class CsstMbiDataManager: ...@@ -325,6 +323,9 @@ class CsstMbiDataManager:
CsstMbiDataManager CsstMbiDataManager
the MBI data manager instance the MBI data manager instance
""" """
# auto identify node name
hostname = os.uname()[1]
if hostname == "dandelion": if hostname == "dandelion":
# dandelion node # dandelion node
dir_l0 = "/nfsdata/share/csst_simulation_data/Cycle-5-SimuData/multipleBandsImaging/" \ dir_l0 = "/nfsdata/share/csst_simulation_data/Cycle-5-SimuData/multipleBandsImaging/" \
...@@ -338,7 +339,7 @@ class CsstMbiDataManager: ...@@ -338,7 +339,7 @@ class CsstMbiDataManager:
path_aux = "/data/sim_data/MSC_0000100/ref/MSC_{}_*_{:02d}_combine.fits" path_aux = "/data/sim_data/MSC_0000100/ref/MSC_{}_*_{:02d}_combine.fits"
dir_pcref = "/home/user/L1Pipeline/msc/gaia_dr3/" dir_pcref = "/home/user/L1Pipeline/msc/gaia_dr3/"
else: else:
raise ValueError("@DM: invalid hostname! valid values are {dandelion, ubuntu}") raise ValueError("@DM: invalid hostname {}!".format(hostname))
return CsstMbiDataManager(ver_sim=ver_sim, dir_l0=dir_l0, dir_l1=dir_l1, dir_pcref=dir_pcref, path_aux=path_aux) return CsstMbiDataManager(ver_sim=ver_sim, dir_l0=dir_l0, dir_l1=dir_l1, dir_pcref=dir_pcref, path_aux=path_aux)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment