Fix a default name in a launch script and slightly improve readability.

PiperOrigin-RevId: 671271512 Change-Id: I1bec7bfd6edf6293293c6a22b3efd12fe99fd20f
google-deepmind · Sep 5, 2024 · 46b06bb · 46b06bb
1 parent fe70bf8
commit 46b06bb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 10 deletions.
diff --git a/examples/modular/launch.py b/examples/modular/launch.py
@@ -55,9 +55,10 @@
 
 # Setup for command line arguments
 parser = argparse.ArgumentParser(description='Run a GDM-Concordia simulation.')
-parser.add_argument(
-    '--agent', action='store', default='basic_agent', dest='agent_name'
-)
+parser.add_argument('--agent',
+                    action='store',
+                    default='basic_agent',
+                    dest='agent_name')
 parser.add_argument('--environment',
                     action='store',
                     default='reality_show',

diff --git a/examples/modular/launch_resident_visitor.py b/examples/modular/launch_resident_visitor.py
@@ -55,15 +55,13 @@
 
 # Setup for command line arguments
 parser = argparse.ArgumentParser(description='Run a GDM-Concordia simulation.')
-parser.add_argument(
-    '--agent_resident',
-    action='store',
-    default='basic_agent',
-    dest='resident_agent_name',
-)
+parser.add_argument('--agent_resident',
+                    action='store',
+                    default='basic_agent',
+                    dest='resident_agent_name')
 parser.add_argument('--agent_visitor',
                     action='store',
-                    default='rational_agent__main_role',
+                    default='rational_agent',
                     dest='visitor_agent_name')
 parser.add_argument('--environment',
                     action='store',