Loading csst_dag/dag/__init__.py +4 −4 Original line number Diff line number Diff line Loading @@ -28,10 +28,10 @@ class CsstDAGs(dict): print(DAG_LIST.keys()) @staticmethod def get_dag(dag: str = ""): def get_dag(self, dag: str = ""): assert dag in DAG_LIST, f"{dag} not in DAG_LIST" return DAG_LIST[dag] return self.get(dag, None) @staticmethod def get_all(): return DAG_LIST def get_all(self): return set(self.keys()) Loading
csst_dag/dag/__init__.py +4 −4 Original line number Diff line number Diff line Loading @@ -28,10 +28,10 @@ class CsstDAGs(dict): print(DAG_LIST.keys()) @staticmethod def get_dag(dag: str = ""): def get_dag(self, dag: str = ""): assert dag in DAG_LIST, f"{dag} not in DAG_LIST" return DAG_LIST[dag] return self.get(dag, None) @staticmethod def get_all(): return DAG_LIST def get_all(self): return set(self.keys())