Commit 77c20b40 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

tweaks

parent 4fd243b6
Loading
Loading
Loading
Loading
+4 −4
Original line number Diff line number Diff line
@@ -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())