fix that random_search can user other datasets

mu_map/random_search/cgan.py

@@ -52,6 +52,7 @@ class cGANRandomSearch(RandomSearch):
             device=device,
             out_dir=out_dir,
             train_dir=train_dir,
+            dataset_dir=dataset_dir,
             logger=logger,
         )
 
@@ -386,5 +387,10 @@ if __name__ == "__main__":
     args = parser.parse_args()
     device = torch.device(args.device)
 
-    random_search = cGANRandomSearch(iterations=args.iterations, device=device, out_dir=args.out_dir, dataset_dir=args.dataset_dir)
+    random_search = cGANRandomSearch(
+        iterations=args.iterations,
+        device=device,
+        out_dir=args.out_dir,
+        dataset_dir=args.dataset_dir,
+    )
     random_search.run()