Convert FOB submodule to regular folder

environments/optimizer/FOB/pytorch_fob/dataset_setup.py

@@ -0,0 +1,22 @@
+import argparse
+from pathlib import Path
+from pytorch_fob.engine.engine import Engine
+
+
+def get_parser():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("experiment_file", type=Path,
+                        help="The yaml file specifying the experiment.")
+    return parser
+
+
+def main(args: argparse.Namespace, extra_args: list[str]):
+    engine = Engine()
+    engine.parse_experiment_from_file(args.experiment_file, extra_args=extra_args)
+    engine.prepare_data()
+
+
+if __name__ == '__main__':
+    parser = get_parser()
+    args, extra_args = parser.parse_known_args()
+    main(args, extra_args)