tweaks

csst_dag/dag/__init__.py

@@ -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())