change create_agent interface

agent/__init__.py

@@ -1 +1 @@
-from .util import create_create_agent
+from .util import create_agent_helper

agent/util.py

@@ -1,12 +1,12 @@
 from recsim.agents import full_slate_q_agent
 
-def create_create_agent(agent=full_slate_q_agent.FullSlateQAgent):
+def create_agent_helper(agent=full_slate_q_agent.FullSlateQAgent, **kwargs):
     def create_agent(sess, environment, eval_mode, summary_writer=None):
-        kwargs = {
-            'observation_space': environment.observation_space,
-            'action_space': environment.action_space,
-            'summary_writer': summary_writer,
-            'eval_mode': eval_mode,
-        }
+        print(f"using {agent.__name__}")
+        kwargs['observation_space'] = environment.observation_space
+        kwargs['action_space'] = environment.action_space
+        kwargs['summary_writer'] = summary_writer
+        kwargs['eval_mode'] = eval_mode
+
         return agent(sess, **kwargs)
     return create_agent