init and __init__

marl_factory_grid/__init__.py

@@ -0,0 +1,6 @@
+from environment.factory import BaseFactory
+from environment.factory import OBSBuilder
+
+from utils.tools import ConfigExplainer
+
+from quickstart import init

marl_factory_grid/quickstart.py

@@ -4,11 +4,16 @@ from pathlib import Path
 
 from marl_factory_grid.utils.tools import ConfigExplainer
 
-if __name__ == '__main__':
+
+def init():
     print('Retrieving available options...')
     ce = ConfigExplainer()
     cwd = Path(os.getcwd())
     ce.save_all(cwd / 'full_config.yaml')
     template_path = Path(__file__) / 'marl_factory_grid' / 'modules' / '_template'
+    print(f'Available config options saved to: {(cwd / "full_config.yaml").resolve()}')
+    print('-----------------------------')
+    print(f'Copying Templates....')
     shutil.copytree(template_path, cwd)
-    print()
+    print(f'Templates copied to {template_path.resolve()}')
+    print(':wave:')