fix bug in DFS

csst_dag/dfs/__init__.py

@@ -60,13 +60,20 @@ class DFS:
             print("Multiple DFS locations are available, please specify one")
             raise ValueError("Multiple DFS locations are available")
         else:
-            self.location = status_table["location"][status_table["status"]][0]
+            if location is None:
+                # set DFS automatically
+                assert (
+                    status_table["status"].sum() == 1
+                ), "Multiple DFS locations are available"
+                location = status_table["location"][status_table["status"]][0]
+                print(f"Using DFS location: {location}")
+            self.location = location
             self.config = CONFIG[self.location]
             print(f"Using DFS location: {location}")
-            for k, v in CONFIG[loc]["dfs"].items():
+            for k, v in CONFIG[self.location]["dfs"].items():
                 os.environ.setdefault(k, str(v))
             # print("Setting redis config:")
-            self.redis = Redis(location=loc)
+            self.redis = Redis(location=self.location)
 
 
 class Redis(redis.Redis):